метрики перекрестной проверки в pyspark.ml - PullRequest
Новая
       4

метрики перекрестной проверки в pyspark.ml

0 голосов
/ 11 октября 2018

Я строю модель перекрестной проверки на наборе данных.Объект перекрестной проверки имеет атрибуты avgMetrics , которые возвращают массив.

evaluator = MulticlassClassificationEvaluator(labelCol='label',predictionCol='prediction',metricName='f1')

crossval = CrossValidator(estimator=self.pipeline,
                          estimatorParamMaps=paramGrid,
                          evaluator=evaluator,
                          numFolds=2)
print(cvModel.avgMetrics)
[0.42175585108783414,
 0.4585199817620358,
 0.39932871646955814,
 0.4310619715036083,
 0.46251606720905303,
 0.42030120869345633,
 0.41120037704551726,
 0.4650393846493733,
 0.3478141848099525]

cvMode.avgMetrics , возвращающий массив метрик.Как мне узнать, какие показатели он возвращает?Есть ли способ получить название метрики?

1 Ответ

0 голосов
/ 11 октября 2018

Это F1 .Вы выбрали его при построении MulticlassClassificationEvaluator

...