@astermind/astermind-synth
Version:
OmegaSynth - Label-Conditioned Synthetic Data Generator for AsterMind ELM/KELM Pipelines
31 lines (30 loc) • 849 B
TypeScript
/**
* 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>;