import { ClassifierBase } from '../base';
import { Params } from '../base/estimator';
import { SubsetSizeOption } from '../utils/paramResolvers';
export interface ExtraTreeProps {
    max_depth?: number;
    min_samples_split?: number;
    criterion?: 'entropy' | 'gini';
    max_features?: SubsetSizeOption;
    randomState?: number;
}
export declare class ExtraTreeClassifier extends ClassifierBase {
    private dtree;
    private max_depth;
    private feature_number;
    private min_samples_split;
    private criterion;
    private max_features_prop?;
    private max_features_;
    private randomState?;
    private random;
    private classesState;
    constructor(props?: ExtraTreeProps);
    getParams(): Params;
    /** Impurity of a class-frequency vector, resolved from `criterion`. */
    private impurity;
    private nodeImpurity;
    private selectFeatures;
    private attributeSelection;
    private treeGenerate;
    fit(sampleX: number[][], sampleY: number[]): void;
    private initTreeNode;
    private findLeaf;
    private resolvedClasses;
    predict(sampleX: number[][]): number[];
    predictProba(sampleX: number[][]): number[][];
    get classes(): number[];
    get featureImportances(): number[];
    getRawFeatureImportances(): number[];
}