使用示例代码Tensorflow不同的形状

时间:2016-10-26 15:36:58

标签: python tensorflow shape

我是张力流的新手,我试图学习它。

我尝试将提供的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

谢谢

0 个答案:

没有答案