cellanet commited on
Commit
9b042db
1 Parent(s): 4cfcfb0

Update machine_learning.py

Browse files
Files changed (1) hide show
  1. machine_learning.py +2 -2
machine_learning.py CHANGED
@@ -173,8 +173,8 @@ output = numpy.array(output)
173
  tensorflow.compat.v1.reset_default_graph()
174
 
175
  net = tflearn.input_data(shape=[None, len(training[0])])
176
- net = tflearn.fully_connected(net, 8)
177
- net = tflearn.fully_connected(net, 8)
178
  net = tflearn.fully_connected(net, len(output[0]), activation="softmax")
179
  net = tflearn.regression(net)
180
 
173
  tensorflow.compat.v1.reset_default_graph()
174
 
175
  net = tflearn.input_data(shape=[None, len(training[0])])
176
+ net = tflearn.fully_connected(net, 32)
177
+ net = tflearn.fully_connected(net, 32)
178
  net = tflearn.fully_connected(net, len(output[0]), activation="softmax")
179
  net = tflearn.regression(net)
180