Decision Tree to If-Else Rules

12 次查看(过去 30 天)
Is there any method to convert directly the decision tree from Classification Tree Viewer to If-Else rules ?
  1 个评论
Mario Buser
Mario Buser 2019-7-9
Would be interested in a solution as well. WEKA, for example, is able to export if/else statements.

请先登录,再进行评论。

回答(1 个)

Koorosh Aslansefat
Koorosh Aslansefat 2020-5-20
I am not sure what exactly you want but the following code can be useful considering that you saved the decision tree as "tc".
CP = tc.CutPoint;
NC = tc.NodeClass;
for ii = 1:size(CP,1)
if ~isnan(CP(ii))
fprintf('if x%d < %f then node %d elseif x%d >= %f then node %d else %d \n',str2num(NC{ii}),CP(ii),ii+1,str2num(NC{ii}),CP(ii), ii+2,ii)
elseif isnan(CP(ii))
fprintf('Class = %d \n',str2num(NC{ii}))
end
end
This equal to veiw(tc)

类别

Help CenterFile Exchange 中查找有关 Statistics and Machine Learning Toolbox 的更多信息

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by