@kanaries/ml

ExtraTreeRegressor

API reference for ExtraTreeRegressor

Tree.ExtraTreeRegressor

interface ExtraTreeRegressorProps {
    max_depth?: number;
    min_samples_split?: number;
    splitter?: 'random';
    max_features?: number | 'sqrt' | 'log2';
}

constructor(props: ExtraTreeRegressorProps = {})

Example

const extra = new ExtraTreeRegressor();
extra.fit(X, Y);
const preds = extra.predict(T);