make_train_test_plots

mastml.plot_helper.make_train_test_plots(run, path, is_classification, label, model, train_X, test_X, groups=None)

General plotting method used to execute sequence of specific plots of train-test data analysis

Args:

run: (dict), a particular split_result from masml_driver

path: (str), path to save the generated plots and analysis of split_result designated in ‘run’

is_classification: (bool), whether or not the analysis is a classification task

label: (str), name of the y data variable being fit

model: (scikit-learn model object), a scikit-learn model/estimator

train_X: (numpy array), array of X features used in training

test_X: (numpy array), array of X features used in testing

groups: (numpy array), array of group names

Returns:

None