1. 程式人生 > >決策樹分類——matlab程式

決策樹分類——matlab程式

%% 使用ID3決策樹演算法預測銷量高低
clc;
clear ;

%% 資料預處理
disp('正在進行資料預處理...');
[matrix,attributes_label,attributes] =  id3_preprocess();

%% 構造ID3決策樹,其中id3()為自定義函式
disp('資料預處理完成,正在進行構造樹...');
tree = id3(matrix,attributes_label,attributes);

%% 列印並畫決策樹
[nodeids,nodevalues] = print_tree(tree);
tree_plot(nodeids,nodevalues);

disp('ID3演算法構建決策樹完成!');