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())