webshap/dist/model/iris-linear.d.ts

Explain any ML models anywhere

/**
 * A logistic regression model for Iris dataset.
 */
export declare class IrisLinearBinary {
    coef: number[];
    intercept: number;
    constructor(coef: number[], intercept: number);
    /**
     * Predict on one data point
     * @param xs Array of data points to predict
     * @returns Predicted classes
     */
    predict: (xs: number[][]) => (1 | 0)[];
    /**
     * Predict on one data point
     * @param xs Array of data points to predict
     * @returns Predicted probabilities
     */
    predictProba: (xs: number[][]) => number[][];
}
/**
 * A logistic regression model for Iris dataset.
 */
export declare class IrisLinearMultiClass {
    coefs: number[][];
    intercepts: number[];
    constructor(coefs: number[][], intercepts: number[]);
    /**
     * Predict on one data point
     * @param xs Array of data points to predict
     * @returns Predicted classes
     */
    predict: (xs: number[][]) => number[];
    /**
     * Predict on one data point
     * @param xs Array of data points to predict
     * @returns Predicted probabilities
     */
    predictProba: (xs: number[][]) => number[][];
}