Изменения

Перейти к: навигация, поиск

Бустинг, AdaBoost

969 байт добавлено, 20:19, 20 января 2019
AdaBoost
Accuracy: 0.9555555555555556
===Пример на языке Scala===
SBT зависимость:
libraryDependencies += "com.github.haifengl" %% "smile-scala" % "1.5.2"
Пример классификации датасета и вычисления F1 меры<ref>[https://en.wikipedia.org/wiki/F1_score F1 мера]</ref> используя smile.classification.adaboost<ref>[https://haifengl.github.io/smile/classification.html#adaboost Smile, AdaBoost]</ref>:
import smile.classification._
import smile.data._
import smile.plot._
import smile.read
import smile.validation.FMeasure
 
val iris: AttributeDataset = read.table("iris.csv", delimiter = ",", response = Some((new NumericAttribute("class"), 2)))
val x: Array[Array[Double]] = iris.x()
val y: Array[Int] = iris.y().map(_.toInt)
val ada: AdaBoost = adaboost(x, y, ntrees = 500, maxNodes = 2)
val predictions: Array[Int] = x.map(ada.predict)
val f1Score = new FMeasure().measure(predictions, y)
plot(x, y, ada)
== См. также ==
Анонимный участник

Навигация