过拟合是指模型在训练数据上表现很好,但在测试数据上表现较差的现象。欠拟合是指模型在训练数据上表现较差,但在测试数据上表现较好的现象。 本题中直接给出了训练数据上的准确率和测试数据上的准确率,所以只需要按照题意进行判断即可。
标准代码如下
def model_fit_quality(training_accuracy, test_accuracy):
"""
基于训练和测试准确率,确定模型是否过拟合、欠拟合或拟合良好,返回1、-1、0。
:param training_accuracy: float, 训练准确率 (0 <= training_accuracy <= 1)
:param test_accuracy: float, 测试准确率 (0 <= test_accuracy <= 1)
:return: int, 1、-1、0
"""
if training_accuracy - test_accuracy > 0.2:
return 1
elif training_accuracy < 0.7 and test_accuracy < 0.7:
return -1
else:
return 0