Cannot get predictions of tensorflow DNNClassifier

前端 未结 5 1548
无人及你
无人及你 2021-01-05 13:24

I\'m using the code from the MNIST tutorial:

feature_columns = [tf.contrib.layers.real_valued_column(\"\", dimension=4)]
classifier = tf.contrib.learn.DNNCla         


        
5条回答
  •  盖世英雄少女心
    2021-01-05 14:25

    Solution:-

    pred = classifier.fit(x=training_set.data, y=training_set.target, steps=2000).predict(test_set.data)
    
    print ("Predictions:")
    
    print(list(pred))
    

    That's it...

提交回复
热议问题