我正在尝试训练DNNClassifier,但是当我尝试训练模型时,我一直遇到称为
AttributeError:模块“ tensorflow.python.framework.tensor_shape”没有属性“ scalar”
import tensorflow as tf
from sklearn.datasets import fetch_openml
mnist = fetch_openml('mnist_784')
feature_columns = [tf.feature_column.numeric_column("x", shape=[784])]
feature_columns
from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(mnist['data'] , mnist['target'])
classifier = tf.estimator.DNNClassifier(hidden_units=[300, 100], n_classes=10,
feature_columns=feature_columns)
def input_train_fn(X,y):
features = {'x':tf.convert_to_tensor(X)}
return features, y.reshape((-1,1)).astype(int)
classifier.train(input_fn=lambda : input_train_fn(X_train, y_train), steps=100000)
最后一行引发以下错误:-
AttributeError:模块“ tensorflow.python.framework.tensor_shape”没有属性“ scalar”
升级tensorflow和tensorflow-estimator可能会有所帮助。
我在Github tf-estimator存储库(https://github.com/tensorflow/estimator/commit/561b3f95d4f9041cb7a2fbbfdc94efee16413725#diff-a815f16c6f90bfe736d98b8c99a7c2fcL337)中找到了此提交。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句