Python 随机森林回归器

我试图使用随机森林回归器，但在rf线上有一条错误消息

from pyspark.ml.regression import RandomForestRegressor
from pyspark.ml.feature import VectorIndexer
from pyspark.ml.feature import VectorAssembler

va = VectorAssembler().setInputCols(diamonds_df_xformed.columns([-4,]).setOutputCol('features')

# Split the data into training and test sets (30% held out for testing)

(trainingData, testData) = diamonds_df_xformed.randomSplit([0.7, 0.3])

# Train a RandomForest model.
rf = RandomForestRegressor(labelCol="price",featuresCol="features")

我得到了一个错误，如图所示，我想知道这个错误消息的问题是什么

---

您没有导入

RandomForestRegressionor

，您已经导入了

RandomForestClassifier

请注意，错误之后出现的任何代码都与问题无关（从未执行），不应包含在此处，因为它只会造成不必要的混乱；对于不相关的导入也是如此。最后，在设置代码格式时要更加小心，否则可能会造成误解，如上面的注释（已编辑）。