UNPKG

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

Version:

Data analysis model package without any dependencies

44 lines (43 loc) 1.25 kB
/** * Self-training */ export default class SelfTraining { /** * @param {object} model View * @param {function (Array<Array<number>>, *[]): void} model.fit Fit model * @param {function (Array<Array<number>>): Array<{category: *, score: number}>} model.predict Returns predicted values * @param {number} threshold Threshold */ constructor(model: { fit: (arg0: Array<Array<number>>, arg1: any[]) => void; predict: (arg0: Array<Array<number>>) => Array<{ category: any; score: number; }>; }, threshold: number); _model: { fit: (arg0: Array<Array<number>>, arg1: any[]) => void; predict: (arg0: Array<Array<number>>) => Array<{ category: any; score: number; }>; }; _threshold: number; /** * Initialize model. * @param {Array<Array<number>>} x Training data * @param {(* | null)[]} y Target values */ init(x: Array<Array<number>>, y: (any | null)[]): void; _x: number[][]; _y: any[]; /** * Fit model. */ fit(): void; /** * Returns predicted categories. * @returns {(* | null)[]} Predicted values */ predict(): (any | null)[]; }