Class

com.github.rzykov.fastml4j.metric

BinaryClassificationMetrics

Related Doc: package metric

Permalink

class BinaryClassificationMetrics extends AnyRef

Calculates binary classification metrics based on a comparison of real and predicted labels

Linear Supertypes
AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. BinaryClassificationMetrics
  2. AnyRef
  3. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new BinaryClassificationMetrics(realLabels: INDArray, predictedLabels: INDArray)

    Permalink

    realLabels

    NDArray of real labels

    predictedLabels

    NDArray of predicted lables

Type Members

  1. case class Confusion(predictedClass: Float, realClass: Float, qty: Int) extends Product with Serializable

    Permalink

    class contains the result of confusion matrix

    class contains the result of confusion matrix

    predictedClass

    predicted class value

    realClass

    real class value

    qty

    number of cases

Value Members

  1. final def !=(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any

  2. final def ##(): Int

    Permalink
    Definition Classes
    AnyRef → Any

  3. final def ==(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any

  4. final def asInstanceOf[T0]: T0

    Permalink
    Definition Classes
    Any

  5. def aucRoc: Float

    Permalink

    calculates AUC value (area under ROC curve)

    calculates AUC value (area under ROC curve)

    returns

    AUC

  6. lazy val binTreshholds: Seq[Float]

    Permalink

    stores unique sorted values of predicted labels

  7. def clone(): AnyRef

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  8. lazy val confusionMatrixByThreshold: Seq[(Float, Seq[Confusion])]

    Permalink

    Calculates confusion matrix for different levels of threshold

  9. final def eq(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef

  10. def equals(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any

  11. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )

  12. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any

  13. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any

  14. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any

  15. final def ne(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef

  16. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef

  17. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef

  18. def precisionByThreshold: Seq[(Float, Float)]

    Permalink

    Calculates Precision metrics for different levels of threshold

    Calculates Precision metrics for different levels of threshold

    returns

    collection of floats with pair: (threshold, precision)

  19. val predictedLabels: INDArray

    Permalink

    NDArray of predicted lables

  20. val realLabels: INDArray

    Permalink

    NDArray of real labels

  21. def recallByTreshold: Seq[(Float, Float)]

    Permalink

    Calculates Recall metrics for different levels of threshold

    Calculates Recall metrics for different levels of threshold

    returns

    collection of floats with pair: (threshold, recall)

  22. def roc: Seq[(Float, Float)]

    Permalink

    Computes the ROC curve

    Computes the ROC curve

    returns

    collection of pairs (fpr, tpr)

  23. final def synchronized[T0](arg0: ⇒ T0): T0

    Permalink
    Definition Classes
    AnyRef

  24. def toString(): String

    Permalink
    Definition Classes
    AnyRef → Any

  25. final def wait(): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  26. final def wait(arg0: Long, arg1: Int): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

  27. final def wait(arg0: Long): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from AnyRef

Inherited from Any

Ungrouped