@ai-on-browser/data-analysis-models/lib/model/label_propagation.d.ts

Data analysis model package without any dependencies

/**
 * Label propagation
 */
export default class LabelPropagation {
    /**
     * @param {'rbf' | 'knn' | { name: 'rbf', sigma?: number, k?: number } | { name: 'knn', k?: number }} [method] Method name
     */
    constructor(method?: "rbf" | "knn" | {
        name: "rbf";
        sigma?: number;
        k?: number;
    } | {
        name: "knn";
        k?: number;
    });
    _affinity: {
        name: "rbf";
        sigma?: number;
        k?: number;
    } | {
        name: "knn";
        k?: number;
    } | {
        name: "rbf" | "knn";
    };
    _affinity_matrix(x: any): Matrix<number>;
    /**
     * Initialize model.
     * @param {Array<Array<number>>} x Training data
     * @param {(* | null)[]} y Target values
     */
    init(x: Array<Array<number>>, y: (any | null)[]): void;
    _y: any[];
    _classes: any[];
    _w: Matrix<number>;
    _probs: Matrix<number>;
    /**
     * Fit model.
     */
    fit(): void;
    /**
     * Returns predicted categories.
     * @returns {*[]} Predicted values
     */
    predict(): any[];
}
import Matrix from '../util/matrix.js';