UNPKG

@ai-on-browser/data-analysis-models

Version:

Data analysis model package without any dependencies

71 lines (70 loc) 2.45 kB
/** * Extreme learning machine classifier */ export class ELMClassifier extends ELM { /** * @param {number[]} size Size of hidden layer * @param {'identity' | 'elu' | 'gaussian' | 'leaky_relu' | 'sigmoid' | 'softplus' | 'softsign' | 'tanh' | function(number): number} [activation] Activation name */ constructor(size: number[], activation?: 'identity' | 'elu' | 'gaussian' | 'leaky_relu' | 'sigmoid' | 'softplus' | 'softsign' | 'tanh' | ((arg0: number) => number)); _classes: any[]; /** * Category list * @type {*[]} */ get categories(): any[]; /** * Fit model. * @param {Array<Array<number>>} x Training data * @param {*[]} y Target values */ fit(x: Array<Array<number>>, y: any[]): void; /** * Returns predicted probabilities. * @param {Array<Array<number>>} x Sample data * @returns {Array<Array<number>>} Predicted values */ probability(x: Array<Array<number>>): Array<Array<number>>; /** * Returns predicted values. * @param {Array<Array<number>>} x Sample data * @returns {*[]} Predicted values */ predict(x: Array<Array<number>>): any[]; } /** * Extreme learning machine regressor */ export class ELMRegressor extends ELM { /** * @param {number[]} size Size of hidden layer * @param {'identity' | 'elu' | 'gaussian' | 'leaky_relu' | 'sigmoid' | 'softplus' | 'softsign' | 'tanh' | function(number): number} [activation] Activation name */ constructor(size: number[], activation?: 'identity' | 'elu' | 'gaussian' | 'leaky_relu' | 'sigmoid' | 'softplus' | 'softsign' | 'tanh' | ((arg0: number) => number)); /** * Fit model. * @param {Array<Array<number>>} x Training data * @param {Array<Array<number>>} y Target values */ fit(x: Array<Array<number>>, y: Array<Array<number>>): void; /** * Returns predicted values. * @param {Array<Array<number>>} x Sample data * @returns {Array<Array<number>>} Predicted values */ predict(x: Array<Array<number>>): Array<Array<number>>; } export type LayerObject = import("./nns/graph").LayerObject; declare class ELM { constructor(size: any, activation: any); _size: any; _activation: any; _a: any; _w: Matrix<number>; _b: Matrix<number>; fit(x: any, y: any): void; _beta: any; predict(x: any): any; } import Matrix from '../util/matrix.js'; export {};