@astermind/astermind-premium
Version:
Astermind Premium - Premium ML Toolkit
45 lines • 1.17 kB
TypeScript
export interface ELMKELMCascadeOptions {
categories: string[];
elmHiddenUnits?: number;
kelmKernel?: 'rbf' | 'polynomial' | 'linear' | 'sigmoid';
kelmGamma?: number;
kelmDegree?: number;
kelmCoef0?: number;
activation?: 'relu' | 'tanh' | 'sigmoid' | 'linear';
maxLen?: number;
useTokenizer?: boolean;
}
export interface ELMKELMCascadeResult {
label: string;
prob: number;
extractedFeatures?: number[];
}
/**
* ELM-KELM Cascade
* Features:
* - ELM for feature extraction
* - KELM for classification
* - Hierarchical learning
* - Efficiency + accuracy
*/
export declare class ELMKELMCascade {
private elm;
private kelm;
private categories;
private options;
private trained;
constructor(options: ELMKELMCascadeOptions);
/**
* Train cascade
*/
train(X: number[][], y: number[] | string[]): void;
/**
* Extract features using ELM
*/
private _extractFeatures;
/**
* Predict with cascade
*/
predict(X: number[] | number[][], topK?: number, returnFeatures?: boolean): ELMKELMCascadeResult[];
}
//# sourceMappingURL=elm-kelm-cascade.d.ts.map