cellanet commited on
Commit
ee2bc44
1 Parent(s): 9379df1

Update machine_learning.py

Browse files
Files changed (1) hide show
  1. machine_learning.py +1 -1
machine_learning.py CHANGED
@@ -143,5 +143,5 @@ if os.path.exists('minh103.meta'):
143
  model.load('minh103')
144
  else:
145
  model = tflearn.DNN(net)
146
- model.fit(training, output, n_epoch=20, batch_size=10, show_metric=True)
147
  model.save('minh103')
 
143
  model.load('minh103')
144
  else:
145
  model = tflearn.DNN(net)
146
+ model.fit(training, output, n_epoch=20000, batch_size=10, show_metric=True)
147
  model.save('minh103')