牛客题解 | 检测过拟合或欠拟合

题目

题目链接

过拟合是指模型在训练数据上表现很好，但在测试数据上表现较差的现象。欠拟合是指模型在训练数据上表现较差，但在测试数据上表现较好的现象。
本题中直接给出了训练数据上的准确率和测试数据上的准确率，所以只需要按照题意进行判断即可。

标准代码如下

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