我需要帮助训练张量流模型。我刚刚开始使用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
谢谢你的帮助。你知道吗
目前没有回答
相关问题 更多 >
编程相关推荐