@astermind/astermind-premium/dist/elm/transfer-learning-elm.d.ts

Astermind Premium - Premium ML Toolkit

export interface TransferLearningELMOptions {
    categories: string[];
    sourceModel?: any;
    freezeBase?: boolean;
    fineTuneLayers?: number;
    hiddenUnits?: number;
    activation?: 'relu' | 'tanh' | 'sigmoid' | 'linear';
    maxLen?: number;
    useTokenizer?: boolean;
}
export interface TransferLearningELMResult {
    label: string;
    prob: number;
}
/**
 * Transfer Learning ELM
 * Features:
 * - Pre-trained model adaptation
 * - Domain adaptation
 * - Few-shot learning
 * - Fine-tuning capabilities
 */
export declare class TransferLearningELM {
    private elm;
    private sourceModel;
    private categories;
    private options;
    private trained;
    constructor(options: TransferLearningELMOptions);
    /**
     * Transfer weights from source model
     */
    private _transferWeights;
    /**
     * Train with transfer learning
     * @param X Target domain features
     * @param y Target domain labels
     */
    train(X: number[][], y: number[] | string[]): void;
    /**
     * Few-shot learning: train with very few examples
     */
    fewShotTrain(X: number[][], y: number[] | string[], shots?: number): void;
    /**
     * Predict with transferred model
     */
    predict(X: number[] | number[][], topK?: number): TransferLearningELMResult[];
    /**
     * Load pre-trained model
     */
    loadSourceModel(model: any): void;
    /**
     * Export current model for use as source in other transfers
     */
    exportModel(): any;
}
//# sourceMappingURL=transfer-learning-elm.d.ts.map