开始使用免费开始使用

模型评估

已经训练好两个模型可供使用:参数较多的 large_model,以及参数较少的 small_model。这两个模型都使用了 train_featurestrain_labels 进行训练,这些变量已为您提供。另有单独的测试集 test_featurestest_labels 可用。

您的目标是比较模型的相对表现,并判断是否存在过拟合迹象。方法是分别在训练集和测试集上评估 large_modelsmall_model。对每个模型,您都可以使用 .evaluate(x, y) 方法,计算特征 x 与标签 y 的损失。然后比较得到的 4 个损失值。

本练习是课程的一部分

Python 中的 TensorFlow 入门

查看课程

练习说明

  • 使用训练数据评估 small 模型。
  • 使用测试数据评估 small 模型。
  • 使用训练数据评估 large 模型。
  • 使用测试数据评估 large 模型。

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Evaluate the small model using the train data
small_train = small_model.evaluate(____, ____)

# Evaluate the small model using the test data
small_test = ____

# Evaluate the large model using the train data
large_train = large_model.evaluate(____, ____)

# Evaluate the large model using the test data
large_test = ____

# Print losses
print('\n Small - Train: {}, Test: {}'.format(small_train, small_test))
print('Large - Train: {}, Test: {}'.format(large_train, large_test))
编辑并运行代码