def add_experiment(name: Text, type: Text, train_data_filename: Text,
test_data_filename: Text, train_data, test_data):
experiment = Experiment(name=name, type=type)
db.session.add(experiment)
db.session.commit()
experiment_locator = Locator(experiment.id,
train_data_filename=train_data_filename,
test_data_filename=test_data_filename)
with open(
os.path.join(experiment_locator.get_train_data_dir(),
train_data_filename), 'wb') as copied_train_data:
for line in train_data.readlines():
copied_train_data.write(line)
with open(
os.path.join(experiment_locator.get_test_data_dir(),
test_data_filename), 'wb') as copied_test_data:
for line in test_data.readlines():
copied_test_data.write(line)
experiment.train_data = train_data_filename
experiment.test_data = test_data_filename
db.session.commit()
return experiment
def test_get_test_data_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/data/test', locator.get_test_data_dir())