更新时间:2023-05-05 GMT+08:00
分享

回归评估

概述

对回归模型预测的结果数据集进行评估。

输入

参数

子参数

参数说明

inputs

dataframe

inputs为字典类型,dataframe为pyspark中的DataFrame类型对象

输出

回归的评估指标:mae、mse、rmse

参数说明

参数

子参数

参数说明

label_col

-

预测结果数据集的目标列的列名

prediction_col

-

预测结果数据集的预测列的列名

样例

inputs = {
    "predict_dataframe": None  # @input {"label":"dataframe","type":"DataFrame"}
}
params = {
    "inputs": inputs,
    "label_col": "",  # @param {"label": "label_col", "type": "string", "required": "true", "helpTip": ""}
    "prediction_col": "prediction"  # @param {"label": "prediction_col", "type": "string", "required": "true", "helpTip": ""}
}
regression_evaluation____id___ = MLSRegressionEvaluation(**params)
regression_evaluation____id___.run()
# @output {"label":"dataframe","name":"regression_evaluation____id___.get_outputs()['output_port_1']","type":"DataFrame"}

分享:

    相关文档

    相关产品