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

Update main.py

parent aa01b2e1
...@@ -40,8 +40,8 @@ def main(): ...@@ -40,8 +40,8 @@ def main():
deepeye_x = np.transpose(trainX, (0, 2, 1)) deepeye_x = np.transpose(trainX, (0, 2, 1))
logging.info(deepeye_x.shape) logging.info(deepeye_x.shape)
deepEye.run(trainX=deepeye_x, trainY=trainY) 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.") 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) RNNdeep.run(trainX=trainX, trainY=trainY)
...@@ -51,4 +51,4 @@ def main(): ...@@ -51,4 +51,4 @@ def main():
logging.info("--- Runtime: %s seconds ---" % (time.time() - start_time)) logging.info("--- Runtime: %s seconds ---" % (time.time() - start_time))
logging.info('Finished Logging') logging.info('Finished Logging')
if __name__=='__main__': if __name__=='__main__':
main() main()
\ No newline at end of file
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