代码:
features = {"lines": np.array(train_lines)}
labels = np.array(train_cat)
train_input_fn = tf.estimator.inputs.numpy_input_fn(
features,
labels,
shuffle=True,
batch_size=16,
num_epochs=20
)
estimator.train(input_fn=train_input_fn)
在运行estimator.train(input_fn = train_input_fn)时遇到此错误
错误:
estimator.train(input_fn = train_input_fn)INFO:tensorflow:正在调用 model_fn。 INFO:tensorflow:调用model_fn。追溯(最新 最后通话):
文件“”,第1行,在 estimator.train(input_fn = train_input_fn)
文件 “ C:\ ProgramData \ Anaconda3 \ lib \ site-packages \ tensorflow_estimator \ python \ estimator \ estimator.py”, 火车上的367号线 损失= self._train_model(input_fn,hooks,saving_listeners)
文件 “ C:\ ProgramData \ Anaconda3 \ lib \ site-packages \ tensorflow_estimator \ python \ estimator \ estimator.py”, _train_model中的第1158行 返回self._train_model_default(input_fn,hooks,saving_listeners)
文件 “ C:\ ProgramData \ Anaconda3 \ lib \ site-packages \ tensorflow_estimator \ python \ estimator \ estimator.py”, 第1188行,在_train_model_default中 功能,标签,ModeKeys.TRAIN,self.config)
文件 “ C:\ ProgramData \ Anaconda3 \ lib \ site-packages \ tensorflow_estimator \ python \ estimator \ estimator.py”, 第1146行,在_call_model_fn中 model_fn_results = self._model_fn(features = features,** kwargs)
文件 “ C:\ ProgramData \ Anaconda3 \ lib \ site-packages \ tensorflow_estimator \ python \ estimator \ canned \ dnn.py”, _model_fn中的第1003行 batch_norm = batch_norm)
文件 “ C:\ ProgramData \ Anaconda3 \ lib \ site-packages \ tensorflow_estimator \ python \ estimator \ canned \ dnn.py”, _dnn_model_fn中的第457行 units = head.logits_dimension,
AttributeError:“列表”对象没有属性“ logits_dimension”