# go-classification

**Repository Path**: stylesmile/go-classification

## Basic Information

- **Project Name**: go-classification
- **Description**: golang 语言实现文本分类 朴素贝叶斯（Naive Bayes）逻辑回归（Logistic Regression）决策树（Decision Tree）
- **Primary Language**: Go
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 0
- **Created**: 2024-06-17
- **Last Updated**: 2024-10-17

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 简介
使用`golang`实现三种分类算法：
* 朴素贝叶斯（Naive Bayes）
* 逻辑回归（Logistic Regression）
* 决策树（Decision Tree）

# 文件说明
* data/ - 来自于UCI数据集文件
* src/data/data.go - 用于将数据集转化为矩阵
* src/classific/decisiontree.go - 决策树算法
* src/classific/naviebayes.go - 朴素贝叶斯算法
* src/classific/logisticregression.go - 逻辑回归算法
* src/classific/classific.go - 指标测试函数F1值,召回率,正确率
* src/main.go - 主函数

# 主函数 - 参数说明
* setnum - K折交叉验证
* classcol - 由data.go转换数据为矩阵，类别所在列号
* alldata - 由data.go转化数据为矩阵

# 测试指标 - 以occupancy.txt数据为例子
进行十折交叉验证，取F1、召回率、正确率平均值：   
| 方法/指标 | F1 | 召回率 | 正确率 |
| --- | --- | --- | --- |
| 朴素贝叶斯 | 0.9667 | 0.9743 | 0.9593 |
| 决策树 | 0.9822 | 0.9909 | 0.9737 | 
| 逻辑回归 | 0.9794 | 0.9886 | 0.9704 |