我正在构建一个TensorFlow模型,该模型应在某些参数或特征超过阈值时输出0或1。我已经有了训练数据集,并且已经训练了模型,但在给定一组数据时,预测结果是错误的。尽管模型的准确率显示为94%,但预测仍然错误。感谢您提前提供的帮助。以下是代码:
回答:
尝试以下更改以使网络更好地训练:
- 缩放输入:您的CSV文件输入未进行缩放。您可以使用类似于sklearn的
StandardScaler
来标准化您的输入 - 训练不足。我看到步数(20)太少。至少尝试10000步,并使用Adam优化器进行更快的收敛:
optimizer = tf.train.AdamOptimizer(learning_rate=0.01)