The Altair Community is migrating to a new platform to provide a better experience for you. In preparation for the migration, the Altair Community is on read-only mode from October 28 - November 6, 2024. Technical support via cases will continue to work as is. For any urgent requests from Students/Faculty members, please submit the form linked here
"Obtaining gain metrics from a decision tree model"
Contributor I
Hi --
Once I've run a DecisionTree learner, is there a way to obtain the attribute selection metric that was computed at each node (e.g. information gain or gain ratio)?
I'm using ModelWriter to output the results, but I don't see that information there.
Also, is there an option to obtain other information such as fraction of data that was split at each node?
Any help much appreciated.
Thanks!
Once I've run a DecisionTree learner, is there a way to obtain the attribute selection metric that was computed at each node (e.g. information gain or gain ratio)?
I'm using ModelWriter to output the results, but I don't see that information there.
Also, is there an option to obtain other information such as fraction of data that was split at each node?
Any help much appreciated.
Thanks!
Tagged:
0
Answers
I am sure (at least with 99%) that is is not possible without digging into the code and alteration of the decision-tree-builder. Beside: It makes no sense that the requested information is stored in the model itself since it is not needed for application.
greetings,
steffen
Steffen is right. This isn't saved in the model at all. Of course one could change it to be saved.
Greetings,
Sebastian