使用與TF .9.0rc0下面上或所以用145個編碼列(1,0)試圖預測1或0用於分類識別記錄..TensorFlowDNNClassifier類已被棄用,但替換似乎不起作用?
classifier_TensorFlow = learn.TensorFlowDNNClassifier(hidden_units=[10, 20, 10],n_classes=2, steps=100)
classifier_TensorFlow.fit(X_train, y_train.ravel())
60,000(火車)和26000(測試)
我得到:
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)
它沒有完成掛起?它不會採用「steps」參數?我沒有得到任何錯誤消息或輸出,所以沒有太多的繼續...任何想法或提示?文檔有點「輕?」
對不起你打的問題,由這似乎是一個錯誤的描述。你可以在https://github.com/tensorflow/tensorflow/issues上提交這個文件嗎? –
對不起,我不能提供完整的測試案例,但大型道具數據 – dartdog