?? form1.cs.txt
字號:
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.Text;
using System.Windows.Forms;
using System.Collections;
using DecisionTree;
namespace DecisionTreeAlgorithm
{
public partial class Form1 : Form
{
enum attributeSelectionMethodType { InformationGain, GainRate, Gini };
enum ageType { year0_20, year21_30, year31_40, year41_50, year51_60, yearover60 };
enum educationType { edu1_3, edu4_6, edu7_9, edu10_12, edu13_14, edu15_16 };
enum occupationType {Tech_support, Craft_repair, Other_service, Sales, Exec_managerial, Prof_specialty,
Handlers_cleaners, Machine_op_inspct, Adm_clerical, Farming_fishing, Transport_moving,
Priv_house_serv, Protective_serv, Armed_Forces};
enum sexType { Male, Female };
enum native_countryType { developNO1, developNO2, developNO3, developNO4, developNO5};
enum makeover50kType { yes, no };
/* 1、 ageType分6類：<=20為year0_20,21-30為year21_30,31-40為year31_40,41-50為year41_50,51-60為year51_60,>=61為yearover60
2、 原有education_numType: 1-16 ,分成6級： edu1_3,edu4_6,edu7_9,edu10_12,edu13_14,edu15_16
3、 occupationType共14類：Tech_support, Craft_repair, Other_service, Sales, Exec_managerial, Prof_specialty,
Handlers_cleaners, Machine_op_inspct, Adm_clerical, Farming_fishing, Transport_moving,
Priv_house_serv, Protective_serv, Armed_Forces
4 、原有native-countryType分成5類developNO1,developNO2,developNO3,developNO4,developNO5后：
developNO1:(<0.1) Outlying-US(Guam-USVI-etc),Vietnam,Mexico,Dominican-Republic,Laos,Haiti,Hungary,
Guatemala,Nicaragua,Scotland,El-Salvador,Trinadad&Tobago,Holand-Netherlands
developNO2:(>=0.1 <0.2) Puerto-Rico,South,China,Cuba,Poland,Jamaica,Portugal,Ireland,Ecuador,Peru,?
developNO3:(>=0.2 <0.3) Honduras,France,Columbia,United-States,England,Germany,Greece,Philippines,Thailand,
Yugoslavia
developNO4:(>=0.3 <0.4) India.Japan.
developNO5:(>=0.4 ) Cambodia,Canada,Iran,Italy,Taiwan,Hong
5、sexType分為Male、Female兩類
*/
struct Item
{
public int ID;
public ageType age;
public educationType education;
public occupationType occupation;
public sexType sex;
public native_countryType native_country;
public makeover50kType makeover50k;
}
private attributeSelectionMethodType attributeSelectionMethod; //屬性選擇度量模式
public ArrayList al; //測試數據
private ArrayList attributeList;
public Form1()
{
InitializeComponent();
}
private void Form1_Load(object sender, EventArgs e)
{
// TODO: 這行代碼將數據加載到表“adult_dataDataSet.traindata2”中。您可以根據需要移動或移除它。
this.traindata2TableAdapter.Fill(this.adult_dataDataSet.traindata2);
// TODO: 這行代碼將數據加載到表“adult_dataDataSet.traindata1”中。您可以根據需要移動或移除它。
this.traindata1TableAdapter.Fill(this.adult_dataDataSet.traindata1);
// TODO: 這行代碼將數據加載到表“adult_dataDataSet.testdata2”中。您可以根據需要移動或移除它。
this.testdata2TableAdapter.Fill(this.adult_dataDataSet.testdata2);
// TODO: 這行代碼將數據加載到表“adult_dataDataSet.testdata1”中。您可以根據需要移動或移除它。
this.testdata1TableAdapter.Fill(this.adult_dataDataSet.testdata1);
attributeSelectionMethod = attributeSelectionMethodType.InformationGain;
al = new ArrayList();
attributeList = new ArrayList();
}
private void databasetoArray()
{
Item testItem;
int rownum = dataGridView1.RowCount;
int i;
al.Clear();
for (i = 0; i < rownum - 1; i++)
{
testItem = new Item();
testItem.ID = al.Count + 1;
testItem.age = (ageType)Enum.Parse(typeof(ageType), dataGridView1.Rows[i].Cells[0].Value.ToString());
testItem.education = (educationType)Enum.Parse(typeof(educationType), dataGridView1.Rows[i].Cells[1].Value.ToString());
testItem.occupation = (occupationType)Enum.Parse(typeof(occupationType), dataGridView1.Rows[i].Cells[2].Value.ToString());
testItem.sex = (sexType)Enum.Parse(typeof(sexType), dataGridView1.Rows[i].Cells[3].Value.ToString());
testItem.native_country = (native_countryType)Enum.Parse(typeof(native_countryType), dataGridView1.Rows[i].Cells[4].Value.ToString());
testItem.makeover50k = (makeover50kType)Enum.Parse(typeof(makeover50kType), dataGridView1.Rows[i].Cells[5].Value.ToString());
al.Add(testItem);
}
}
private void infomationGainToolStripMenuItem_Click(object sender, EventArgs e)
{
this.infomationGainToolStripMenuItem.Checked = true;
this.gainRateToolStripMenuItem.Checked = false;
this.giniToolStripMenuItem.Checked = false;
attributeSelectionMethod = attributeSelectionMethodType.InformationGain;
CreateTree();
}
private void gainRateToolStripMenuItem_Click(object sender, EventArgs e)
{
this.infomationGainToolStripMenuItem.Checked = false;
this.gainRateToolStripMenuItem.Checked = true;
this.giniToolStripMenuItem.Checked = false;
attributeSelectionMethod = attributeSelectionMethodType.GainRate;
CreateTree();
}
private void giniToolStripMenuItem_Click(object sender, EventArgs e)
{
this.infomationGainToolStripMenuItem.Checked = false;
this.gainRateToolStripMenuItem.Checked = false;
this.giniToolStripMenuItem.Checked = true;
attributeSelectionMethod = attributeSelectionMethodType.Gini;
CreateTree();
}
private void CreateTree()
{
if (istraindata2 == false ) //確保選擇的是“清理后的訓練數據集”
{
MessageBox.Show("請先輸入清理后的訓練數據集,并重新選擇決策樹度量屬性");
testtextBox.Text = "";
}
else
{
switch (attributeSelectionMethod)
{
case attributeSelectionMethodType.InformationGain:
break;
case attributeSelectionMethodType.GainRate:
break;
case attributeSelectionMethodType.Gini:
break;
default:
MessageBox.Show("請選擇正確的決策樹度量屬性");
break;
}
this.DecisionTree.Nodes.Clear();
testtextBox.Text = "";
attributeList.Clear();
attributeList.Add("age");
attributeList.Add("education");
attributeList.Add("occupation");
attributeList.Add("sex");
attributeList.Add("native_country");
this.DecisionTree.Nodes.Add(GenerateDecisionTree(al, attributeList));
}
}
//算法：Generate_decision_tree。由數據劃分D的訓練元組產生決策樹。
private TreeNode GenerateDecisionTree(ArrayList d, ArrayList attribute_list)
{
//(1) 創建一個節點N
TreeNode temp = new TreeNode();
//(2) ifD中的元組都是同一類C then
string C = FindSameClass(d);
if (C != null)
//(3) 返回N作為葉節點，以類C標記
{
temp.Text = C;
return temp;
}
//(4) if attribute_list為空 then
if (attribute_list.Count == 0)
//(5) 返回N作為葉子節點，標記為D中的多數類；//多數表決
{
temp.Text = FindMoreClass(d);
return temp;
}
//(6) 使用attribute_selection_method(D,attribute_list)，找出“最好”的splitting_criterion
SplitCriterion splittingCriterion = AttributeSelectionMethod(d, attribute_list);
//(7) 用splitting_criterion標記節點N
temp.Text = splittingCriterion.attribute;
//(8) if splitting_attribute是離散值的并且允許多路劃分 then //不限于二叉樹
if (true)//假定總是離散的并且允許多路劃分
//(9) attribute_list<--attribute_list-splitting_attribute //刪除劃分屬性
{
attribute_list.Remove(splittingCriterion.attribute);
}
//(10) for splitting_criterion的每個輸出j //劃分元組并對每個劃分產生子樹
for (int i = 0; i < splittingCriterion.splitList.Count; i++)
{
//(11) 設Dj是D中滿足輸出j的數據元組的集合 //一個劃分
ArrayList Dj = (ArrayList)splittingCriterion.splitList[i];
//(12) if Dj為空 then
if (Dj.Count == 0)
//(13) 加一個樹葉到節點N，標記為D中的多數類
{
TreeNode leaf = new TreeNode();
leaf.Text = FindMoreClass(d);
temp.Nodes.Add(leaf);
}
//(14) else 加一個由 Generate_decision_tree(Dj,attribute_list)返回的節點到節點N
else
{
TreeNode leaf = GenerateDecisionTree(Dj, new ArrayList(attribute_list));
leaf.Text = splittingCriterion.descriptionList[i].ToString() + "&&" + leaf.Text;
temp.Nodes.Add(leaf);
}
// end for
}
//(15) 返回N
return temp;
}
//尋找D中的同一類C，找到返回C的名稱，找不到返回NULL。
private string FindSameClass(ArrayList d)
{
Item temp = new Item();
bool find;
find = true;
for (int i = 0; i < d.Count; i++)
{
if (i == 0)
{
temp.makeover50k = ((Item)d[i]).makeover50k;
}
else
{
if (temp.makeover50k != ((Item)d[i]).makeover50k)
{
find = false;
break;
}
}
}
if (find) return " => makeover50k=" + temp.makeover50k.ToString();
return null;
}
//尋找D中的多數類
private string FindMoreClass(ArrayList d)
{
int count = 0;
string temp = " => makeover50k=";
for (int i = 0; i < d.Count; i++)
{
if (((Item)d[i]).makeover50k == makeover50kType.yes)
count++;
}
if (2 * count >= d.Count)
return temp + "yes";
else
return temp + "no";
}
//找出“最好”的splitting_criterion
private SplitCriterion AttributeSelectionMethod(ArrayList d, ArrayList attribute_list)
{
SplitCriterion splitting_criterion = new SplitCriterion();
switch (attributeSelectionMethod)
{
case attributeSelectionMethodType.InformationGain:
//double info; //按類劃分的期望信息
double infoA; //按A劃分的期望信息
//double gainA; //信息增益
//計算info
/*
info = Info(d);
*/
//計算最小的InfoA和選擇的屬性
infoA = 1;
?? 快捷鍵說明
復制代碼
Ctrl + C
搜索代碼
Ctrl + F
全屏模式
F11
切換主題
Ctrl + Shift + D
顯示快捷鍵
?
增大字號
Ctrl + =
減小字號
Ctrl + -