#python #machine-learning #pyspark
#python #машинное обучение #pyspark
Вопрос:
Я пытался использовать регрессор случайного леса, но в строке 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")
Я получил сообщение об ошибке, как показано на графике, интересно, в чем проблема в этом случае сообщение об ошибке.
Комментарии:
1. вы не импортировали
RandomForestRegressor, вы импортировалиRandomForestClassifier2. Пожалуйста, обратите внимание, что любой код, который появляется после ошибки, не имеет отношения к проблеме (никогда не выполняется), и его не следует включать сюда, поскольку он просто создает ненужный беспорядок; то же самое справедливо и для нерелевантного импорта. Наконец, будьте более осторожны с форматированием кода, иначе вы можете создать недоразумения, как в комментарии выше (отредактировано).