1. Nauka
  2. /
  3. Kursy
  4. /
  5. Podstawy PySpark

Connected

ćwiczenie

Tworzenie ewaluatora

Pierwszą rzeczą potrzebną przy krosswalidacji do wyboru modelu jest sposób porównywania różnych modeli. Na szczęście submoduł pyspark.ml.evaluation zawiera klasy do oceny różnych rodzajów modeli. Twój model to model klasyfikacji binarnej, dlatego skorzystasz z klasy BinaryClassificationEvaluator z modułu pyspark.ml.evaluation.

Ten ewaluator oblicza pole pod krzywą ROC. Jest to metryka łącząca dwa rodzaje błędów, jakie może popełnić klasyfikator binarny (fałszywe pozytywne i fałszywe negatywne), w jedną prostą liczbę. Więcej na ten temat dowiesz się pod koniec rozdziału!

Instrukcje

100 XP
  • Zaimportuj submoduł pyspark.ml.evaluation jako evals.
  • Utwórz zmienną evaluator, wywołując evals.BinaryClassificationEvaluator() z argumentem metricName="areaUnderROC".