def train(cls, experiment): name = experiment.name exp_locator = Locator(experiment.id, experiment.train_data, experiment.test_data) if 'LR_' in name: lr_dataset_maker = LRPimaIndiansDatasetMaker(exp_locator) X, y = lr_dataset_maker.make_train_dataset() hyperparams = {'penalty': 'l2'} logistic_regression = Logistic_Regression(hyperparams=hyperparams) result = logistic_regression.train(X, y) DataLoader.save(file_object=logistic_regression, file_path=os.path.join(exp_locator.get_model_dir(), 'model.pkl')) experiment.result = json.dumps([result]) return experiment raise Exception("No valid Experiment Name to train Experiment")
def test_get_model_dir(self): """Ensure get_train_data_dir behaves properly""" locator = Locator(id=1, train_data_filename='pima_indians.csv', test_data_filename='pima_indians.csv') self.assertIn('/experiments/1/model', locator.get_model_dir())