UNPKG

@astermind/astermind-elm

Version:

JavaScript Extreme Learning Machine (ELM) library for browser and Node.js.

24 lines (23 loc) 1.17 kB
import { ELMConfig } from '../core/ELMConfig'; export declare class FeatureCombinerELM { private elm; private config; private categories; constructor(config: ELMConfig & { inputSize?: number; }); /** Concatenate encoder vector + metadata vector */ static combineFeatures(encodedVec: number[], meta: number[]): number[]; /** Convenience for batch combination */ static combineBatch(encoded: number[][], metas: number[][]): number[][]; /** Train from encoder vectors + metadata + labels (classification) */ train(encoded: number[][], metas: number[][], labels: string[]): void; /** Predict top-K labels from a single (vec, meta) pair */ predict(encodedVec: number[], meta: number[], topK?: number): import("../core/ELM").PredictResult[]; /** Predict the single best label + prob */ predictLabel(encodedVec: number[], meta: number[]): import("../core/ELM").PredictResult; /** Get hidden embedding for (vec, meta) pair (useful for chaining) */ getEmbedding(encodedVec: number[], meta: number[]): number[]; loadModelFromJSON(json: string): void; saveModelAsJSONFile(filename?: string): void; }