Home Reference Source
import {Estimator} from '@jsmlt/jsmlt/src/supervised/base.js'
public class | source

Estimator

Direct Subclass:

Classifier

Base class for supervised estimators (classifiers or regression models).

Method Summary

Public Methods
public abstract

predict(X: Array<Array<number>>): Array<mixed>

Make a prediction for a data set.

public abstract

train(X: Array<Array<number>>, y: Array<mixed>)

Train the supervised learning algorithm on a dataset.

Public Methods

public abstract predict(X: Array<Array<number>>): Array<mixed> source

Make a prediction for a data set.

Params:

NameTypeAttributeDescription
X Array<Array<number>>

Features for each data point

Return:

Array<mixed>

Predictions. Label of class with highest prevalence among k nearest neighbours for each sample

public abstract train(X: Array<Array<number>>, y: Array<mixed>) source

Train the supervised learning algorithm on a dataset.

Params:

NameTypeAttributeDescription
X Array<Array<number>>

Features per data point

y Array<mixed>

Class labels per data point