attr'TI'的DataType字符串不在允许值列表中：uint8，int32，int64

Question

我一直在使用CNN进行文本分类并使用tensorflow的contrib learn。

然而，当我尝试执行以下代码：

classifier = learn.Estimator(model_fn=cnn_model) 

classifier.fit(x_train, y_train, steps=10000) 
y_predicted = [ p['class'] for p in classifier.predict(x_test, as_iterable=True)] 

score = metrics.accuracy_score(y_test, y_predicted) 

print('Accuracy: {0:f}'.format(score)) 

我下面的错误运行：

ERROR:DataType string for attr 'TI' not in list of allowed values: uint8, int32, int64 on line 'classifier.fit'

Answer 1

您需要将输入转换y_train给定类型。 print(type(y_train))最有可能是一个浮点数，而不是一个整数。