nctstuti commited on
Commit
dd51fbc
·
verified ·
1 Parent(s): 8318dbe

Upload RecurrentPPO model for WQI prediction

Browse files
Files changed (4) hide show
  1. README.md +33 -3
  2. RecurrentPPO.png +0 -0
  3. metadata.json +1 -0
  4. rppo_wqi_model.zip +3 -0
README.md CHANGED
@@ -1,3 +1,33 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ library_name: stable-baselines3
4
+ tags:
5
+ - stable-baselines3
6
+ - reinforcement-learning
7
+ - RecurrentPPO
8
+ - water-quality-prediction
9
+ ---
10
+
11
+ # RecurrentPPO Model for Water Quality Index Prediction
12
+
13
+ This model uses RecurrentPPO with an LSTM policy to predict Water Quality Index (WQI) values.
14
+
15
+ ## Model Description
16
+
17
+ - Model Type: RecurrentPPO
18
+ - Policy: MlpLstmPolicy
19
+ - Training Steps: 10000
20
+ - Average MAPE: 22.08%
21
+
22
+ ## Usage
23
+
24
+ ```python
25
+ from sb3_contrib import RecurrentPPO
26
+
27
+ # Load the model
28
+ model = RecurrentPPO.load("nctstuti/ds_btl")
29
+
30
+ # Make predictions
31
+ obs = your_observation # Shape should match your feature space
32
+ action, _states = model.predict(obs)
33
+ ```
RecurrentPPO.png ADDED
metadata.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"model_type": "RecurrentPPO", "policy_type": "MlpLstmPolicy", "features": ["Place", "pH", "DO", "\u0110\u1ed9 d\u1eabn", "N-NO2", "N-NH4", "P-PO4", "TSS", "COD", "Aeromonas t\u1ed5ng s\u1ed1"], "target": "WQI", "metrics": {"total_steps": 10, "average_mape": 22.08037726170723}}
rppo_wqi_model.zip ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c46cbebc8ccc6b4f05544b05566509695cd0b93f28dbb97f9c46be0a0d7920f
3
+ size 7121935