Класс 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)

это висит без завершения? это не будет принимать параметр «шаги»? Я не получаю сообщений об ошибках или вывода, так что не так много, чтобы продолжить ... Есть идеи или советы? Документация немного "легкая?"

Ответы на вопрос(1)

Ваш ответ на вопрос