diff options
| author | Vasil Zlatanov <v@skozl.com> | 2019-03-05 14:32:38 +0000 |
|---|---|---|
| committer | Vasil Zlatanov <v@skozl.com> | 2019-03-05 14:32:38 +0000 |
| commit | 6b573a30a3021d259400af9751645eb1a5b4705b (patch) | |
| tree | 09d68aa13d19062369f5044bf246ddd89a579fdb /lenet.py | |
| parent | 740e1b0c6a02a7bec20008758373f0dd80baade4 (diff) | |
| parent | 2a720c237259baa2d968286244f9e43794c7e4d9 (diff) | |
| download | e4-gan-6b573a30a3021d259400af9751645eb1a5b4705b.tar.gz e4-gan-6b573a30a3021d259400af9751645eb1a5b4705b.tar.bz2 e4-gan-6b573a30a3021d259400af9751645eb1a5b4705b.zip | |
Full featured VBN
Diffstat (limited to 'lenet.py')
| -rw-r--r-- | lenet.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -125,8 +125,8 @@ def test_classifier(model, x_test, y_true): y_pred = model.predict(x_test) y_pred = np.argmax(y_pred, axis=1) y_true = np.argmax(y_true, axis=1) - print("Test acc:", accuracy_score(y_true, y_pred)) plot_example_errors(y_pred, y_true, x_test) + return accuracy_score(y_true, y_pred) def mix_data(X_train, y_train, X_validation, y_validation, train_gen, tr_labels_gen, val_gen, val_labels_gen, split=0): |