The bostonpredict's intro from zzsqwq

bostonpredict's Introduction

BostonPredict

此仓库用来对波士顿问题进行预测，使用sklearn中的公开数据集，不过在data里面也附录了一份数据集，基于pytorch实现最终的模型。

对应博客地址为：利用神经网络进行波士顿房价预测 - Zs's blog

最终效果如下：

Network Sructure

网络是一个二层的前馈神经网络，结构图大致如下：

Requirements

需要Python 3.6 及以上版本，低版本可能也能用，需要大家自己尝试，在仓库根目录中使用如下命令来安装requirements.txt 中的所有的依赖

$ pip install -r requirements.txt

Predict

仓库中主要包含两个文件：predict.py 和 Regression.py

predict.py

实现了利用前馈神经网络对波士顿房价进行预测，包含训练和预测两部分，里面有部分可选参数，具体如下：

$ python predict.py -h

usage: predict.py [-h] [--weights WEIGHTS] [--load_weights]
                  [--hidden_layer HIDDEN_LAYER] [--learn_rate LEARN_RATE]
                  [--input_shape INPUT_SHAPE]
                  [--load_cols LOAD_COLS [LOAD_COLS ...]] [--epoch EPOCH]

optional arguments:
  -h, --help            show this help message and exit
  --weights WEIGHTS     inital weights path
  --load_weights        load weights or not
  --hidden_layer HIDDEN_LAYER
                        The dim of hidden_layer
  --learn_rate LEARN_RATE
                        The learning rate
  --input_shape INPUT_SHAPE
                        The input_shape of networks,don't forget change
                        load_cols
  --load_cols LOAD_COLS [LOAD_COLS ...]
  --epoch EPOCH         The epoch of train