@astermind/astermind-elm
Version:
JavaScript Extreme Learning Machine (ELM) library for browser and Node.js.
24 lines (23 loc) • 1.17 kB
TypeScript
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;
}