• 功能介绍
  • 参数说明
  • 脚本示例
    • 脚本代码

    功能介绍

    本组件用One VS Rest策略进行多分类。

    参数说明

    名称 中文名称 描述 类型 是否必须？ 默认值
    numClass 类别数 多分类的类别数，必选 Integer ✓
    predictionCol 预测结果列名 预测结果列名 String ✓
    predictionDetailCol 预测详细信息列名 预测详细信息列名 String
    reservedCols 算法保留列名 算法保留列 String[] null

    脚本示例

    脚本代码

    1. URL = "http://alink-dataset.cn-hangzhou.oss.aliyun-inc.com/csv/iris.csv";
    2. SCHEMA_STR = "sepal_length double, sepal_width double, petal_length double, petal_width double, category string";
    3. data = CsvSourceBatchOp().setFilePath(URL).setSchemaStr(SCHEMA_STR)
    5. lr = LogisticRegression() \
    6.     .setFeatureCols(["sepal_length", "sepal_width", "petal_length", "petal_width"]) \
    7.     .setLabelCol("category") \
    8.     .setMaxIter(100)
    10. oneVsRest = OneVsRest().setClassifier(lr).setNumClass(3)
    11. model = oneVsRest.fit(data)
    12. model.setPredictionCol("pred_result").setPredictionDetailCol("pred_detail")
    13. model.transform(data).print()