前言

本文基于 Kaggle平臺—洪水數據集的回歸預測（文末附數據集），更新CNN、LSTM、LSTM-Attention、Transformer-BiLSTM、CNN-BiLSTM-Attention等模型的可視化分析！

1.更新介紹（新增可視化代碼）

1.1 新增可視化對比

（1）柱狀圖對比：

（2）雷達圖可視化對比：

（3）預測擬合對比：

1.2 模型簡介

包括完整流程數據代碼處理：

回歸預測數據集制作、數據加載、模型定義、參數設置、模型訓練、模型測試、預測可視化、模型評估

2.數據預處理

數據集格式為CSV文件，共50000個樣本，20個特征，來預測FloodProbability（洪水概率）：　該結果變量基于上述因素預測洪水的可能性，可能表示為0到1之間的概率。

按照7:2:1劃分訓練集、驗證集、測試集：

3.基于CNN-BiLSTM-Attention的回歸模型

3.1 定義CNN-BiLSTM-Attention網絡模型

3.2 設置參數，訓練模型

50個epoch，mse極低，CNN-BiLSTM-Attention網絡效果顯著，模型能夠充分提取數據的多尺度特征，收斂速度快，性能特別優越，效果明顯。

4 模型評估與可視化

4.1 模型評估

4.2 回歸預測擬合

圖片

本文轉載自???建模先鋒???，作者：小蝸愛建模