UNPKG

@astermind/astermind-premium

Version:

Astermind Premium - Premium ML Toolkit

56 lines 1.38 kB
export interface ConvolutionalELMOptions { categories: string[]; inputShape?: number[]; filters?: number[]; kernelSizes?: number[]; poolSizes?: number[]; hiddenUnits?: number; activation?: 'relu' | 'tanh' | 'sigmoid' | 'linear'; maxLen?: number; useTokenizer?: boolean; } export interface ConvolutionalELMResult { label: string; prob: number; featureMaps?: number[][]; } /** * Convolutional ELM * Features: * - Convolutional layers for feature extraction * - ELM for classification * - Translation invariance * - Image/sequence processing */ export declare class ConvolutionalELM { private elm; private categories; private options; private trained; constructor(options: ConvolutionalELMOptions); /** * Train on image/sequence data */ train(X: number[][][] | number[][], y: number[] | string[]): void; /** * Extract features using convolutional layers */ private _extractConvolutionalFeatures; /** * Convolutional layer (simplified) */ private _convLayer; /** * Pooling layer */ private _poolLayer; /** * Flatten feature maps */ private _flatten; /** * Predict */ predict(X: number[][][] | number[][], topK?: number): ConvolutionalELMResult[]; } //# sourceMappingURL=convolutional-elm.d.ts.map