master
ml server 2020-01-03 23:42:31 -07:00
parent 3945f5f826
commit 7b9fdc4e33
1 changed files with 2 additions and 0 deletions

2
wut-ml
View File

@ -54,6 +54,8 @@ model.compile(loss='binary_crossentropy',
optimizer='rmsprop',
metrics=['accuracy'])
model.fit(x=train_it, validation_data=val_it, epochs=1, verbose=2, workers=8, use_multiprocessing=True)
prediction = model.predict(x=test_it, batch_size=None, verbose=0, steps=None, use_multiprocessing=True)
print(prediction)