import { LayersModel, Tensor } from '@tensorflow/tfjs-node';
import { ModelEvaluation } from './types';
export declare class ModelManager {
    readonly model: LayersModel;
    constructor(model: LayersModel);
    train(x: Tensor, y: Tensor, args: {
        epochs: number;
        batchSize: number;
    }): Promise<void>;
    evaluate(x: Tensor, y: Tensor): Promise<ModelEvaluation>;
    predict(x: Tensor): Tensor;
    save(path: string): Promise<void>;
}
