import { ClassifierBase } from '../base';
import { Params } from '../base/estimator';
import { type NumericMatrix } from '../data';
export interface CategoricalNBProps {
    alpha?: number;
    forceAlpha?: boolean;
    fitPrior?: boolean;
    classPrior?: number[] | null;
    minCategories?: number | number[] | null;
}
export declare class CategoricalNB extends ClassifierBase {
    readonly acceptedInputKinds: readonly ["dense", "csr"];
    private alpha;
    private forceAlpha;
    private fitPrior;
    private classPrior;
    private minCategories;
    private classes;
    private classCount;
    private categoryCount;
    private classLogPrior;
    private featureLogProb;
    private nCategories;
    constructor(props?: CategoricalNBProps);
    getParams(): Params;
    private initCounters;
    fit(trainX: NumericMatrix, trainY: number[]): void;
    predict(testX: NumericMatrix): number[];
}