Commit b07cd8f0 authored by Ard Kastrati's avatar Ard Kastrati
Browse files

Merge branch 'master' of https://gitlab.ethz.ch/kard/dl-project

parents f0ade3d4 f72aa634
......@@ -19,8 +19,7 @@ sns.set_style('darkgrid')
def run(trainX, trainY):
if config['split']:
config['model']=config['model']+'_cluster'
classifier = Classifier_DEEPEYE_LSTM(output_directory=config['root_dir'],
input_shape=config['deepeye-lstm']['input_shape'])
classifier = Classifier_DEEPEYE_LSTM(output_directory=config['root_dir'], input_shape=config['deepeye-lstm']['input_shape'])
hist = classifier.fit(trainX, trainY)
plot_loss(hist, config['model_dir'], config['model'], True)
plot_acc(hist, config['model_dir'], config['model'], True)
......
......@@ -40,8 +40,8 @@ def main():
deepeye_x = np.transpose(trainX, (0, 2, 1))
logging.info(deepeye_x.shape)
deepEye.run(trainX=deepeye_x, trainY=trainY)
elif config['model'] == 'deepeye-lstm':
elif config['model'] == 'deepeye-lstm':
logging.info("Started running deepeye-lstm. If you want to run other methods please choose another model in the config.py file.")
RNNdeep.run(trainX=trainX, trainY=trainY)
......@@ -51,4 +51,4 @@ def main():
logging.info("--- Runtime: %s seconds ---" % (time.time() - start_time))
logging.info('Finished Logging')
if __name__=='__main__':
main()
\ No newline at end of file
main()
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment