Commit 7809754e authored by zigeng's avatar zigeng
Browse files

Update main.py

parent aa01b2e1
......@@ -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