declare module 'ml-logistic-regression' {
    import type Matrix from 'ml-matrix';

    class LogisticRegressionTwoClasses {
        testScores(Xtest: Matrix): number;
    }

    export default class LogisticRegression {
        classifiers: LogisticRegressionTwoClasses[];

        constructor(
            options: Partial<{
                numSteps: number;
                learningRate: number;
            }>,
        );

        train(X: Matrix, Y: Matrix): void;

        predict(Xtest: Matrix): number[];

        static load(model: Record<string, unknown>): LogisticRegression;

        toJSON(): Record<string, unknown>;
    }
}
