# 机器学习作业

**Repository Path**: volturihao/homework1

## Basic Information

- **Project Name**: 机器学习作业
- **Description**: 杭研院研一下高飞老师机器学习大小作业代码库
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2024-05-13
- **Last Updated**: 2024-05-21

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 项目结构

```
├─data  存放fashion-mnist数据集
├─cnn.py 第二次作业的CNN代码
├─gpu_info.py 查看自己的cuda信息，便于配置GPU加速
├─mnist_reader.py 提供读取fashion-mnist数据集的方法 
├─svm.py  第一次作业的svm代码
├─requirements.txt 依赖列表
├─.gitignore git忽略列表，不用管
```



# 环境配置

直接在环境命令行中使用以下命令安装依赖即可：

```
pip install -r requirements.txt
```

其中tensorflow版本需要根据自己的cuda版本进行决定，否则无法启用gpu加速，但是这点训练量不启用也问题不大

## 如何启用gpu加速：

若没安装cuda则根据tensorflow版本安装cuda

若有cuda则直接在anaconda环境下运行以下命令：

```
conda install tensorflow-gpu
```

会自动升降级到适合的tensorflow版本



## 代码运行

第一次作业直接运行`svm.py`即可

第二次作业若要启用gpu加速，先运行`gpu_info.py`，根据gpu信息修改`cnn.py`中以下代码即可：

```python
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
```

若不启用gpu加速直接注释掉上面这行代码即可