@astermind/astermind-synth
Version:
OmegaSynth - Label-Conditioned Synthetic Data Generator for AsterMind ELM/KELM Pipelines
33 lines (32 loc) • 1.07 kB
TypeScript
/**
* Utilities for saving trained OmegaSynth models
*/
import { OmegaSynth } from '../OmegaSynth';
import { LabeledSample } from '../types';
export interface SavedModelData {
config: {
mode: string;
maxLength?: number;
seed?: number;
exactMode?: boolean;
useOneHot?: boolean;
useClassification?: boolean;
usePatternCorrection?: boolean;
};
trainingStats: {
totalSamples: number;
labels: string[];
samplesPerLabel: Record<string, number>;
};
timestamp: string;
}
/**
* Save a trained OmegaSynth model to disk
*
* @param synth The trained OmegaSynth instance
* @param trainingData The training data used to train the model (required for saving)
* @param outputDir Directory where the model will be saved
* @param version Optional version string (default: '1.0.0')
* @returns Path to the saved model directory
*/
export declare function saveTrainedModel(synth: OmegaSynth, trainingData: LabeledSample[], outputDir: string, version?: string): Promise<string>;