Sparklyr |多类Logistic回归问题
我们试图在Sparklyr中拟合多类Logistic回归模型。数据集使用“ mtcars”,我们试图预测的目标变量是“ gear”,它具有3类,方法“ ml_logistic_regression”引发以下错误“错误:对象281的方法areaUnderROC错误:对象java.lang.IllegalArgumentException:无效”代码对于具有2个类的目标变量“ am”可以很好地工作
sc <-spark_connect(master =“ yarn-client”,version = spark_version,config = config,spark_home = spark_home)
mtcars_tbl <-sdf_copy_to(sc,mtcars,name =“ mtcars_tbl”,覆盖= TRUE)
lr_model <-mtcars_tbl%>%ml_logistic_regression(am〜gear + carb)
lr_model <-mtcars_tbl%>%ml_logistic_regression(齿轮〜am + carb,family =“多项式”)