predictor not working
aliraza0337 opened this issue · 1 comments
aliraza0337 commented
I am trying to run the train.py in '/workload/prediction' its throwing error,
NameError: name 'nb_epoch' is not defined.
What should I define 'nb_epoch' as?
marcoszh commented
Hi Ali,
Thanks for the interest in our work.
It is the number of epochs to train the lstm network for, you can tweak it yourself it to suit your accuracy need.