我是张力流的新手,我试图学习它。
我尝试将提供的Iris示例(here)与我自己的数据一起使用。
我有这个CSV:
2262,2013,2452,2260,2279,2340,2160,2404,5162,6259,4882,2544,2365,2293,2333,2283,2063,2156,2427,2305,1
2316,2423,2350,2150,2241,2167,2086,3109,4312,6150,4761,2837,2464,2135,2455,2339,2417,2281,2016,2284,1
...
依此类推(21列45行,最后一行是类别)。
我只有2个类别(0和1),我使用此代码
training_set = tf.contrib.learn.datasets.base.load_csv_without_header(filename="data.csv",
target_dtype=numpy.int,features_dtype=numpy.int,target_column=20)
# Specify that all features have real-value data
feature_columns = [tf.contrib.layers.real_valued_column("",dimension=20)]
# Build 3 layer DNN with 10, 20, 10 units respectively.
classifier = tf.contrib.learn.DNNClassifier(feature_columns=feature_columns,
hidden_units=[10, 20, 10],
n_classes=2,
model_dir="/tmp/24-10")
classifier.fit(x=training_set.data, y=training_set.target, steps=2000)
我得到的错误是关于形状,但我似乎没有发现此代码有什么问题
错误(相对于拟合指令在线):
ValueError: Shapes (?, 1) and (?,) are incompatible
谢谢