@astermind/astermind-synth
Version:
OmegaSynth - Label-Conditioned Synthetic Data Generator for AsterMind ELM/KELM Pipelines
47 lines (46 loc) • 1.67 kB
TypeScript
/**
* loadPretrained - Load pretrained synthetic data generator
* Instantiates OmegaSynth with pretrained data for common labels
*/
import { OmegaSynth } from '../OmegaSynth';
export interface PretrainedModel {
version: string;
labels: Record<string, string[]>;
}
/**
* Load pretrained OmegaSynth instance
* @param mode Generation mode ('retrieval', 'elm', or 'hybrid')
* @param config Optional configuration overrides
*/
export declare function loadPretrained(mode?: 'retrieval' | 'elm' | 'hybrid', config?: {
maxLength?: number;
seed?: number;
}): OmegaSynth;
/**
* Load a fully versioned OmegaSynth model from dist/models/vX.Y.Z
*
* This function:
* - Reads model.json, training_data.json, and elm_model.json from the version directory
* - Rebuilds the retrieval store from training_data.json
* - Hydrates the internal ELM from elm_model.json (for elm/hybrid modes) if possible
*
* NOTE:
* - We avoid calling synth.train() here to prevent re-training; instead we:
* - Directly ingest training samples into the retrieval generator
* - Attempt to load ELM weights via loadModelFromJSON if available
*/
export declare function loadPretrainedFromVersion(versionDir: string): OmegaSynth;
/**
* Load pretrained model from custom JSON data
* @param modelData Custom model data
* @param mode Generation mode
* @param config Optional configuration
*/
export declare function loadPretrainedFromData(modelData: PretrainedModel, mode?: 'retrieval' | 'elm' | 'hybrid', config?: {
maxLength?: number;
seed?: number;
}): OmegaSynth;
/**
* Get available pretrained labels
*/
export declare function getPretrainedLabels(): string[];