next htofix
This commit is contained in:
parent
c94d5f815e
commit
095d8aead2
@ -31,6 +31,6 @@ def regression_model():
|
|||||||
return model
|
return model
|
||||||
|
|
||||||
model = regression_model()
|
model = regression_model()
|
||||||
model.fit(x_train, y_train, batch_size=int(sys.argv[1]), epochs = sys.argv[2]) #verbose = 1
|
model.fit(x_train, y_train, batch_size=int(sys.argv[1]), epochs = int(sys.argv[1])) #verbose = 1
|
||||||
|
|
||||||
model.save('wine_model')
|
model.save('wine_model')
|
Loading…
Reference in New Issue
Block a user