spacecruft
/
satnogs-wut
0
0
Fork 0
Code Issues 6 Releases Activity

softmax

master
ml server 2020-01-04 13:35:09 -07:00
parent 2707edb725
commit 8c2640d88d
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
wut-ml
View File

@ -49,12 +49,13 @@ model.add(Dense(64))
model.add(Activation('relu'))
model.add(Dropout(0.5))
model.add(Dense(1))
model.add(Activation('sigmoid'))
model.add(Activation('softmax'))
#model.compile(loss='categorical_crossentropy',
model.compile(loss='binary_crossentropy',
optimizer='rmsprop',
metrics=['accuracy'])
model.fit(x=train_it, validation_data=val_it, epochs=16, verbose=2, workers=16, use_multiprocessing=True)
model.fit(x=train_it, validation_data=val_it, epochs=2, verbose=2, workers=16, use_multiprocessing=True)
prediction = model.predict(x=test_it, batch_size=None, verbose=0, steps=None, use_multiprocessing=True)
print(prediction)