[Rapidminer]How to Get Rules from a Decision Tree?

是 否 +2 论坛币

k人 参与回答

经管之家送您一份

应届毕业生专属福利!

求职就业群

赵安豆老师微信：zhaoandou666

经管之家联合CDA

送您一个全额奖学金名额~ !

立即领取

感谢您参与论坛问题回答

经管之家送您两个论坛币！

+2 论坛币

How to Get Rules from a Decision Tree?by mschmitz on 10-21-2016 09:53 AM - edited on 11-03-2016 07:35 AM by stevefarr


Question

Decision Trees are well known for their understandability. How can I get the rules a Decision Tree provides in a more handy format?

Answer

To convert Decision Trees into Rules you can use the Tree to Rule operator. The Tree to Rule operator is a nested operator, means you can put the Decision Tree inside.

The Decision Tree is parsed into a rule format which is easier to understand

This Rule Model can be parsed into a example set with the following Groovy script. The Groovy script needs to be used in an Execute Script operator. An example is attached as a process.

1. import com.rapidminer.operator.learner.rules.RuleModel;
   
2. import com.rapidminer.tools.Ontology;
   
3. import com.rapidminer.tools.LogService;
   
4. import com.rapidminer.operator.learner.rules.Rule;
   
5. 
   
6. import java.util.logging.Level
   
7. RuleModel ruleModel = input[0]
   
8. numberOfAttributes = 4;
   
9. 
   
10. Attribute[] attributes= new Attribute[numberOfAttributes];
    
11. attributes[0] = AttributeFactory.createAttribute("Full Rule", Ontology.STRING);
    
12. attributes[1] = AttributeFactory.createAttribute("Label", Ontology.STRING);
    
13. attributes[2] = AttributeFactory.createAttribute("Correct Examples covered by this rule", Ontology.STRING);
    
14. attributes[3] = AttributeFactory.createAttribute("Wrong Examples covered by this rule", Ontology.STRING);
    
15. MemoryExampleTable table = new MemoryExampleTable(attributes);
    
16. DataRowFactory ROW_FACTORY = new DataRowFactory(0);
    
17. 
    
18. String[] myvalues = new String[numberOfAttributes]
    
19. 
    
20. for(Rule currentRule : ruleModel.getRules()){
    
21.     int correct = 0;
    
22.     int wrong = 0;
    
23.     int label = ruleModel.getLabel().getMapping().getIndex(currentRule.getLabel());
    
24.     LogService.root.log(Level.INFO, currentRule.toString())
    
25.     int[] frequencies = currentRule.getFrequencies();
    
26.     if (frequencies != null) {
    
27.         for (int i = 0; i < frequencies.length; i++) {
    
28.             if (i == label) {
    
29.                 correct += frequencies[i];
    
30.             } else {
    
31.                 wrong += frequencies[i];
    
32.           }
    
33.       }
    
34.       myvalues[0] = currentRule.toString()
    
35.       myvalues[1] = currentRule.getLabel()
    
36.       myvalues[2] = String.valueOf(correct);
    
37.       myvalues[3] = String.valueOf(wrong);
    
38. 
    
39.       DataRow row = ROW_FACTORY.create(myvalues, attributes)
    
40.       table.addDataRow(row);
    
41.     }
    
42. }
    
43. 
    
44. return table.createExampleSet();

复制代码

The result looks like the screen shot below:

扫码加我 拉你入群

请注明：姓名-公司-职位

以便审核进群资格，未注明则拒绝

分享0 收藏0 回帖

关键词：RapidMiner Decision DMINE Rapid Rules understand Community following provides convert