diff options
author | Vasil Zlatanov <v@skozl.com> | 2019-03-10 19:00:54 +0000 |
---|---|---|
committer | Vasil Zlatanov <v@skozl.com> | 2019-03-10 19:00:54 +0000 |
commit | f9c0139f63438a2574d9931f732cb3aecd172485 (patch) | |
tree | 5b17163ce054ad8113f6ee265c4b377a1df52dc8 | |
parent | 07ae3a62f460e8b1ae700194cd85f05b2d2f011d (diff) | |
download | e4-gan-f9c0139f63438a2574d9931f732cb3aecd172485.tar.gz e4-gan-f9c0139f63438a2574d9931f732cb3aecd172485.tar.bz2 e4-gan-f9c0139f63438a2574d9931f732cb3aecd172485.zip |
Fix y_true in plot_probas
-rw-r--r-- | lenet.py | 1 |
1 files changed, 1 insertions, 0 deletions
@@ -144,6 +144,7 @@ def train_classifier(x_train, y_train, x_val, y_val, batch_size=128, epochs=100, return model def plot_probas(model, x_test, y_true): + y_true = np.argmax(y_true, axis=1) x_test = np.pad(x_test, ((0,0),(2,2),(2,2),(0,0)), 'constant') probas = model.predict(x_test) skplt.metrics.plot_roc(y_true, probas) |