8. 要点总结

训练机器学习模型的步骤包括:

制定任务:

  • 是回归问题还是分类问题?
  • 可以通过监督式学习还是非监督式学习来完成?
  • 输入数据的形状是什么?输出数据应该是什么样的?

准备数据:

  • 清理数据并尽可能手动检查它是否存在任何模式
  • 在使用数据进行训练之前对数据进行重排
  • 将数据归一化为神经网络的合理范围。通常,对于数值数据,0-1 或 -1-1 是合适的范围。
  • 将数据转换为张量

构建并运行您的模型:

  • 使用 tf.sequential 或 tf.model 定义模型,然后使用 tf.layers.* 向模型中添加层
  • 选择优化器(adam 通常是一个不错的选择),以及批次大小和周期数等参数。
  • 为您的问题选择合适的损失函数,并选择准确率指标来帮助您评估进度。meanSquaredError 是处理回归问题的常见损失函数。
  • 监控训练,看看损失是否降低

评估模型

  • 为您的模型选择一个评估指标,您可以在训练过程中对模型进行监控。训练完成后,请尝试进行一些测试预测,以了解预测质量。