import { Dataset } from '@tensorflow/tfjs-data';
import { TensorContainer } from '@tensorflow/tfjs-core';
import { default as Model, ModelForwardAttributes } from '../models/model';
import { Conversation, ITokeniser } from '../main';
interface Result {
    loss: number;
    accuracy: number;
}
export default class Evaluator {
    private model;
    private iterator?;
    private xs?;
    private ys?;
    private masked;
    constructor(model: Model<ModelForwardAttributes>, dataset: Dataset<TensorContainer> | Conversation[][], tokeniser?: ITokeniser, masked?: boolean);
    dispose(): void;
    private calculateBatchLoss;
    evaluate(maxBatches?: number): Promise<Result | Result[]>;
}
export {};
