import { ITokeniser, Task } from '../main';
import { Tensor } from '@tensorflow/tfjs-core';
import { Dataset } from '@tensorflow/tfjs-data';
import { DatasetBuilder, DatasetState } from './DatasetBuilder';
export declare function createTrainValidationSplit(tasks: Task[] | Uint16Array, tokeniser: ITokeniser, datasetBuilder: DatasetBuilder, batchSize: number, validationSplit?: number): Promise<{
    trainDataset: Dataset<{
        xs: Tensor;
        ys: Tensor;
    }>;
    validationDataset: Dataset<{
        xs: Tensor;
        ys: Tensor;
    }>;
    size: number;
    validationState: DatasetState;
    trainState: DatasetState;
}>;
