TypeError:传递给参数“ref”的值的数据类型int64不在允许值列表中:float32、int32、qint8、quint8、qint32

2024-04-24 14:42:12 发布

您现在位置:Python中文网/ 问答频道 /正文

我使用Raspberry pi3,在安装带有https://github.com/samjabrahams/tensorflow-on-raspberry-pi/blob/master/GUIDE.md#3-build-bazel指示的Tensorflow之后,我尝试用Python运行一个脚本。在

Python version : Python 2.7.9

tensorflow._ version _ : '1.5.0-rc0'

我的代码:

# FEATURES, LABEL and COLUMNS are defined 

def get_input_fn(data_set, num_epochs=None, shuffle=True):
      return tf.estimator.inputs.pandas_input_fn(
  x=pd.DataFrame({k: data_set[k].values for k in FEATURES}),
  y = pd.Series(data_set[LABEL].values),
  num_epochs=num_epochs,
  shuffle=shuffle)

# ...

data_train_np = np.zeros((data_train_size,7),dtype=np.float)
# Creation of data_train_np values ...
data_train=pd.DataFrame(data_train_np,columns=COLUMNS)

# ...

feature_cols = [tf.feature_column.numeric_column(k) for k in FEATURES] 
regressor = tf.estimator.DNNRegressor(feature_columns=feature_cols,hidden_units=[256,64,32])
regressor.train(input_fn=get_input_fn(data_train), steps=100)

运行脚本后,我有:

^{pr2}$

我看到一些问题有类似的错误信息,但我的案例不知道如何解决它。在

谢谢你的回答!在


Tags: inputdatatftensorflownptrainnumfeature