LogisticRegression

public final class LogisticRegression implements Classifier

Constructors

Link copied to clipboard
public LogisticRegression LogisticRegression(GradientDescent optimizer)

Functions

Link copied to clipboard
public Unit fit(MultiArray<Float, D2> X, NDArray<Integer, D2> y)
Link copied to clipboard
public NDArray<Integer, D2> predict(MultiArray<Float, D2> X)
Link copied to clipboard
public final MultiArray<Float, D2> predictProba(MultiArray<Float, D2> X)