@astermind/astermind-elm
Version:
JavaScript Extreme Learning Machine (ELM) library for browser and Node.js.
29 lines (28 loc) • 826 B
TypeScript
import type { EncoderLike } from './ELMChain';
import { ELM } from './ELM';
import { OnlineELM } from './OnlineELM';
type OnlineMode = 'hidden' | 'logits';
export type AdapterTarget = {
type: 'elm';
model: ELM;
name?: string;
} | {
type: 'online';
model: OnlineELM;
name?: string;
mode?: OnlineMode;
};
export declare class ELMAdapter implements EncoderLike {
readonly name: string;
private readonly target;
private readonly mode;
constructor(target: AdapterTarget);
/** Return embeddings for a batch (N x D) -> (N x H/L) */
getEmbedding(X: number[][]): number[][];
}
export declare function wrapELM(model: ELM, name?: string): ELMAdapter;
export declare function wrapOnlineELM(model: OnlineELM, opts?: {
name?: string;
mode?: OnlineMode;
}): ELMAdapter;
export {};