24cc9536e4969716b3080498426af8e47dee7907,tests/classifiers/test_tensorflow.py,TestTFClassifier,test_predict,#TestTFClassifier#,59

Before Change

    def test_predict(self):
        // Get MNIST
        (_, _), (x_test, y_test) = self.mnist

        // Test fit and predict
        preds = self.classifier.predict(x_test)

After Change

        sess.close()

    def test_predict(self):
        classifier, sess = get_classifier_tf()

        predictions = classifier.predict(self.x_test)
        predictions_class = np.argmax(predictions, axis=1)
        trues_class = np.argmax(self.y_test, axis=1)

Instances