Skip to content

Commit

Permalink
fixed test script
Browse files Browse the repository at this point in the history
  • Loading branch information
mikkokotila committed Jul 25, 2018
1 parent ccc855f commit 7fdb0c9
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions test_script.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,9 +7,10 @@
from keras.layers import Dropout, Dense

from keras.optimizers import SGD, Adam, Adadelta, Adagrad, Adamax, RMSprop, Nadam
from keras.activations import softmax, relu, elu, sigmoid
from keras.activations import softmax, relu, sigmoid
from keras.losses import categorical_crossentropy, logcosh, binary_crossentropy
from talos.metrics.keras_metrics import matthews_correlation, precision, recall, fmeasure
from talos.metrics.keras_metrics import matthews_correlation
from talos.metrics.keras_metrics import precision, recall, fmeasure_acc


def iris_model(x_train, y_train, x_val, y_val, params):
Expand Down Expand Up @@ -63,7 +64,7 @@ def cervix_model(x_train, y_train, x_val, y_val, params):
model.compile(optimizer=params['optimizer'](lr=lr_normalizer(params['lr'], params['optimizer'])),
loss=params['loss'],
metrics=['acc',
fmeasure,
fmeasure_acc,
recall,
precision,
matthews_correlation])
Expand Down

0 comments on commit 7fdb0c9

Please sign in to comment.