import * as tf from '@tensorflow/tfjs'; export interface IClasses { [index: string]: number; } export interface IData { classes: IClasses; [index: string]: IImageData; } export interface IImageData { xs?: tf.Tensor; ys?: tf.Tensor2D; } export interface ICollectedData { classes: IClasses; xs?: tf.Tensor; ys?: tf.Tensor2D; } export interface IParams { [index: string]: any; batchSize?: number; epochs?: number; } export declare type TypedArray = Int8Array | Uint8Array | Int16Array | Uint16Array | Int32Array | Uint32Array | Uint8ClampedArray | Float32Array | Float64Array; export interface IArgs { pretrainedModel?: string | tf.Model; trainingModel?: tf.Model | Function; onLoadStart?: Function; onLoadComplete?: Function; onAddDataStart?: Function; onAddDataComplete?: Function; onClearDataStart?: Function; onClearDataComplete?: Function; onTrainStart?: Function; onTrainComplete?: Function; onPredictComplete?: Function; onPredictStart?: Function; onEvaluateStart?: Function; onEvaluateComplete?: Function; onSaveStart?: Function; onSaveComplete?: Function; }