/NIRATSpecNet

0

Primary LanguageJupyter Notebook

代码实现

  • 2022/10/28更新

添加上了somefunction.py,该文件用户从数据集读取数据以及简单计算一些数据 添加了return_model.py文件,可用于返回一个基线模型 关于论文复现,由于模型变量初始化参数不同,会导致结果有浮动,代码中已给出带权重的h5模型文件,加载该模型文件即可复现出论文结果


  • test下为数据集

  • .ipyb 后缀文件是PLS、SVR和ANN代码

  • .png 后缀文件是可视化之后的原始结果

  • Model.py 文件可用来加载CY.h5(柴油模型)、YM.h5(玉米模型)和NN(牛奶模型)等

  • regulating_Hyperparameter.py 是模型结构

  • 随机初始化会影响模型结果,使用我训练好的模型即可复现论文结果(后缀为.h5 的模型),尽量使用下文所示环境

  • 硬件环境 操作系统 windows10、显卡 Tesla V100 32G

  • 软件环境 python==3.6 、Keras==2.3 、Tensorflow==2.0