DT
所属分类:数值算法/人工智能
开发工具:Python
文件大小:26350KB
下载次数:1
上传日期:2020-12-31 04:16:21
上 传 者:
Aliyu
说明: Full Python implementation of a decision tree classifier with pruning
文件列表:
Data (0, 2017-08-27)
Data\train_in.csv (76335889, 2017-08-27)
Data\train_out.csv (1050153, 2017-08-27)
Src (0, 2017-08-27)
Src\decision_tree.py (10824, 2017-08-27)
# arXiv Decision Tree Classifier
Full Python implementation of a decision tree classifier with pruning. Classifies arXiv abstracts by topic.
Sample usage:
```
python decision_tree.py -h
python decision_tree.py -i ../Data/train_in.csv -o ../Data/train_out.csv -f 400 -e 1.5 -s 1000
```
A decision tree will be trained on `train_in.csv` and evaluated against `train_out.csv`. It will use 400 words as features and have an entropy threshold of 1.5, it will use 1000 examples to train the tree.
Will produce a file `DTmetrics.txt` containing the performance metrics of the decision tree classifier.
近期下载者:
相关文件:
收藏者: