UNPKG

@astermind/astermind-elm

Version:

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

29 lines (28 loc) 826 B
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 {};