@astermind/astermind-premium
Version:
Astermind Premium - Premium ML Toolkit
56 lines • 1.38 kB
TypeScript
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