TypeError: Tlabels属性的数据类型字符串不在允许的值列表中:int32、int64。

2024-04-23 08:21:31 发布

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

我需要帮助训练张量流模型。我刚刚开始使用Google云平台,并试图用自己的数据集来训练自己的模型。你知道吗

下面是我目前的代码

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import tensorflow as tf
import numpy as np
import h5py

# Datasets
training_set1 = h5py.File('/Users/Fang/workspace/keras_project/for_google-ml/trainer/11_type.h5', 'r')
output_set1 = open("/Users/Fang/workspace/keras_project/for_google-ml/trainer/output_type.txt", "r")

X = training_set1['the_data'][:]
Y = output_set1.read().split(',')

feature_columns = [tf.contrib.layers.real_valued_column("", dimension=2)]

classifier = tf.contrib.learn.DNNClassifier(feature_columns=feature_columns,
                                        hidden_units=[500, 100, 11],
                                        n_classes=3,
                                        model_dir="/tmp/the_model")

classifier.fit(X, Y, steps=2000)
accuracy_score = classifier.evaluate(X, Y)["accuracy"]
print("Accuracy: %.2f".format(accuracy_score))

我在本地运行时遇到了这样的错误。你知道吗

TypeError: DataType string for attr 'Tlabels' not in list of allowed values: int32, int64

我的training_set由以下以hdf5格式保存为the_data数据集的数据组成。它有多个数组,每个数组有500个数字。示例:

[[12.424, 384.742,...],
 [3492.293, 349,..,...],
 [...,...,...],
  ...
  ...
 [...]]

我的output_set是一个包含以下500个数据的文本文件

aaa,aaa,aaa,...,bbb,bbb,bbb,...ccc,ccc,...,kkk,kkk,kkk

谢谢你的帮助。你知道吗


Tags: columnsthe数据fromimportforoutputtf