更新时间:2024-04-29 GMT+08:00
分享

多分类评估

概述

对多分类模型预测的结果数据集进行评估。

输入

参数

子参数

参数说明

inputs

dataframe

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

输出

多分类的评估指标:准确率、混淆矩阵

参数说明

参数

子参数

参数说明

label_col

-

目标列

prediction_index_col

-

输入预测数据集的预测label标签列的列名

label_index_col

-

输入预测数据集的真实label标签列的列名

样例

inputs = {
    "predict_dataframe": None  # @input {"label":"dataframe","type":"DataFrame"}
}
params = {
    "inputs": inputs,
    "label_col": "",  # @param {"label": "label_col", "type": "string", "required": "true", "helpTip": ""}
    "prediction_index_col": "prediction_index",  # @param {"label": "prediction_index_col", "type": "string", "required": "true", "helpTip": ""}
    "label_index_col": "label_index"  # @param {"label": "label_index_col", "type": "string", "required": "true", "helpTip": ""}
}
multi_class_evaluation____id___ = MLSMultiClassEvaluation(**params)
multi_class_evaluation____id___.run()
# @output {"label":"dataframe","name":"multi_class_evaluation____id___.get_outputs()['output_port_1']","type":"DataFrame"}

分享:

    相关文档

    相关产品