Class DecisionTreeClassifier

Decision tree classifier

Hierarchy

DecisionTree
- DecisionTreeClassifier

Index

Constructors

Properties

_datas _depth _features _method _tree

Accessors

Methods

_calcScore _calcValue _classesRate _gini _id3 fit importance init predict predict_prob predict_value

Constructors

constructor

new DecisionTreeClassifier(method: "ID3" | "CART"): DecisionTreeClassifier
Parameters
- method: "ID3" | "CART"
  Method name
Returns DecisionTreeClassifier
Overrides DecisionTree.constructor
- Defined in lib/model/decision_tree.d.ts:8

Properties

_datas

_datas: { target: any; value: number[] }[]

_depth

_depth: number

_features

_features: number

_method

_method: "ID3" | "CART"

_tree

_tree: {
    children: any[];
    datas: { target: any; value: number[] }[];
    leafs: any;
    score: any;
    value: any;
}

Accessors

depth

get depth(): number
Depth of the tree

Returns number
Inherited from DecisionTree.depth
- Defined in lib/model/decision_tree.d.ts:50

Methods

_calcScore

_calcScore(datas: any): number
Parameters
- datas: any
Returns number
- Defined in lib/model/decision_tree.d.ts:11

_calcValue

_calcValue(datas: any): Map<any, any>
Parameters
- datas: any
Returns Map<any, any>
- Defined in lib/model/decision_tree.d.ts:10

_classesRate

_classesRate(datas: any): Map<any, any>
Parameters
- datas: any
Returns Map<any, any>
- Defined in lib/model/decision_tree.d.ts:12

_gini

_gini(datas: any): number
Parameters
- datas: any
Returns number
- Defined in lib/model/decision_tree.d.ts:14

_id3

_id3(datas: any): number
Parameters
- datas: any
Returns number
- Defined in lib/model/decision_tree.d.ts:13

fit

fit(): void
Fit model.

Returns void
Inherited from DecisionTree.fit
- Defined in lib/model/decision_tree.d.ts:75

importance

importance(): number[]
Returns importances of the features.

Returns number[]
Importances
Inherited from DecisionTree.importance
- Defined in lib/model/decision_tree.d.ts:80

init

init(datas: number[][], targets: any[]): void
Initialize model.
Parameters
- datas: number[][]
  Training data
- targets: any[]
  Target values
Returns void
Inherited from DecisionTree.init
- Defined in lib/model/decision_tree.d.ts:56

predict

predict(data: number[][]): any[]
Returns predicted values.
Parameters
- data: number[][]
  Sample data
Returns any[]
Predicted values
- Defined in lib/model/decision_tree.d.ts:26

predict_prob

predict_prob(data: number[][]): number[]
Returns probability of the datas.
Parameters
- data: number[][]
  Sample data
Returns number[]
Predicted values
- Defined in lib/model/decision_tree.d.ts:20

predict_value

predict_value(data: number[][]): number[]
Returns predicted values.
Parameters
- data: number[][]
  Sample data
Returns number[]
Predicted values
Inherited from DecisionTree.predict_value
- Defined in lib/model/decision_tree.d.ts:86