Windows 10
PyTorch 1.2.0
python 3.5.4
利用神经网络对文本数据集分类。
data.txt
simple_network.py
首先确保数据集的格式和data.txt中的格式一致，即特征+标签。如果不一致，需要更改simple_network.py中的load_data()函数。其最终返回值为特征和标签。
在main中依次选择训练和测试部分即可。
利用CNN对图片数据集分类。
dataSet
load_data.py
info.yml
model.py
train.py
predict.py
首先打开info.yml，更改里面的配置信息。
数据集加载采用DataLoader模式，返回可迭代的数据集和标签序列。
采用普通的卷积，池化和全连接层。
在main中依次选择训练和测试部分即可。