scikit_learn_test.py | 49 def assert_classification_works(clf): 57 clf.fit(x_train, y_train, batch_size=BATCH_SIZE, epochs=EPOCHS) 59 score = clf.score(x_train, y_train, batch_size=BATCH_SIZE) 62 preds = clf.predict(x_test, batch_size=BATCH_SIZE) 67 proba = clf.predict_proba(x_test, batch_size=BATCH_SIZE) 106 clf = keras.wrappers.scikit_learn.KerasClassifier( 112 assert_classification_works(clf) 122 clf = keras.wrappers.scikit_learn.KerasClassifier( variable in class:ScikitLearnAPIWrapperTest.test_classify_class_build_fn.ClassBuildFnClf 128 assert_classification_works(clf) 138 clf = InheritClassBuildFnClf variable in class:ScikitLearnAPIWrapperTest.test_classify_inherit_class_build_fn.InheritClassBuildFnClf [all...] |