pyspark databricks中分类逻辑回归模型的标签无效

Question

我正在使用 Spark ML 库通过逻辑回归来解决分类问题。

我已经对输入特征进行了矢量化处理，并创建了训练数据集和测试数据集。

在拟合模型时，我遇到了无效标签问题。

训练数据集是：

我的输入特征为Independent_features，我的目标特征为Category_con。

Answer 1

在创建向量时使用标签、特征而不是 Independent_features 和 Category_con。