import { SGDClassifier } from './SgdClassifier'; import { ModelFitArgs } from '../types'; export interface LogisticRegressionParams { /** Specify the norm of the penalty. **default = l2** */ penalty?: 'l1' | 'l2' | 'none'; /** Inverse of the regularization strength. **default = 1** */ C?: number; /** Whether or not the intercept should be estimator not. **default = true** */ fitIntercept?: boolean; modelFitOptions?: Partial; } /** Builds a linear classification model with associated penalty and regularization * * @example * ```js * let X = [ [1, -1], [2, 0], [2, 1], [2, -1], [3, 2], [0, 4], [1, 3], [1, 4], [1, 5], [2, 3], ] let y = [ 0, 0, 0, 0, 0, 1, 1, 1, 1, 1] let logreg = new LogisticRegression({ penalty: 'none' }) await logreg.fit(X, y) * ``` */ export declare class LogisticRegression extends SGDClassifier { constructor({ penalty, C, fitIntercept, modelFitOptions }?: LogisticRegressionParams); }