wut fit
parent
3945f5f826
commit
7b9fdc4e33
2
wut-ml
2
wut-ml
|
@ -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)
|
||||
|
||||
|
|
Loading…
Reference in New Issue