Tensorflow predicts always the same result

后端 未结 2 2050
小蘑菇
小蘑菇 2020-12-06 14:52

I\'m trying to get the TensorFlow example running with my own data, but somehow the classifier always picks the same class for every test example. The input data is always s

2条回答
  •  长情又很酷
    2020-12-06 15:25

    The other problem you may be having is a Class imbalance. If you have one class that greatly outweighs the other your function may be converging to that value. Try balancing the classes in your training sample as well as using smaller batches. For example if your labels are binary then make sure there is an equal amount of zeros and one labels in your training sample.

提交回复
热议问题