UNPKG

@astermind/astermind-synth

Version:

OmegaSynth - Label-Conditioned Synthetic Data Generator for AsterMind ELM/KELM Pipelines

31 lines (30 loc) 849 B
/** * Train OmegaSynth model on combined training data */ import { OmegaSynth } from '../OmegaSynth'; import { LabeledSample } from '../types'; export interface TrainingConfig { mode: 'retrieval' | 'elm' | 'hybrid'; maxLength?: number; seed?: number; trainingFiles: string[]; } export interface TrainedModel { synth: OmegaSynth; config: TrainingConfig; trainingStats: { totalSamples: number; labels: string[]; samplesPerLabel: Record<string, number>; }; elmModelJSON?: string; trainingSamples: LabeledSample[]; } /** * Train a model on the specified training files */ export declare function trainModel(config: TrainingConfig): Promise<TrainedModel>; /** * Save trained model to file */ export declare function saveModel(model: TrainedModel, outputDir: string): Promise<string>;