Home > Enterprise >  Information gain for decision tree in Weka
Information gain for decision tree in Weka

Time:10-25

I am using GainRatioAttributeEval in Weka. I got this result

enter image description here

I can not interpret this result. I mean what does this portion mean?

Ranked attributes:
 1.0000000000000009    5 Na_to_K
 0.3935404073616423    3 BP
 0.2063964793752467    1 Age
 0.09316668988826002   4 Cholesterol
 0.007712386387024157  2 Sex

My dataset has 6 attributes with 200 instances. Among them drug is my class attribute. I am also attaching the preprocess overview of the dataset.

enter image description here

I know how to calculate information gain and create a decision tree. But I can not get this result.

CodePudding user response:

Gain ratio and info gain are two separate attribue evaluation methods with different formulas. See the linked Javadoc for more information.

  • Related