Класс TensorFlowDNNClassifier устарел, но замена не работает?
Используя следующее с TF .9.0rc0 на 60 000 (поезд) и 26 000 (тест) на или около того записях с 145 кодированными столбцами (1,0), пытаясь предсказать 1 или 0 для идентификации класса.
classifier_TensorFlow = learn.TensorFlowDNNClassifier(hidden_units=[10, 20, 10],n_classes=2, steps=100)
classifier_TensorFlow.fit(X_train, y_train.ravel())
Я получил:
WARNING:tensorflow:TensorFlowDNNClassifier class is deprecated. Please consider using DNNClassifier as an alternative.
Out[34]:TensorFlowDNNClassifier(steps=100, batch_size=32)
И тогда хорошие результаты довольно быстро:
score = metrics.accuracy_score(y_test, classifier_TensorFlow.predict(X_test))
print('Accuracy: {0:f}'.format(score))
Accuracy: 0.923121
А также:
print (metrics.confusion_matrix(y_test, X_pred_class))
[[23996 103]
[ 1992 15]]
Но когда я пытаюсь использовать новый предложенный метод:
classifier_TensorFlow = learn.DNNClassifier(hidden_units=[10, 20, 10],n_classes=2)
это висит без завершения? это не будет принимать параметр «шаги»? Я не получаю сообщений об ошибках или вывода, так что не так много, чтобы продолжить ... Есть идеи или советы? Документация немного "легкая?"