LabelPropagation
API reference for LabelPropagation
SemiSupervised.LabelPropagation
interface LabelPropagationOptions {
kernel?: 'rbf' | 'knn' | ((X: number[][], Y: number[][]) => number[][]);
gamma?: number;
nNeighbors?: number;
maxIter?: number;
tol?: number;
}
constructor(options: LabelPropagationOptions = {})Label propagation assigns labels to unlabeled data by propagating information from labeled points across a graph defined by a kernel.
Methods
fit(trainX: number[][], trainY: number[]): voidpredict(testX: number[][]): number[]predictProba(testX: number[][]): number[][]
Example
const lp = new LabelPropagation();
lp.fit(trainX, trainY);
const preds = lp.predict(testX);