adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

callback function and TDQN model added

Browse Source
This commit is contained in:
MukavaValkku
2022-08-13 18:48:58 +03:00
committed by robcaulk
parent 8eeaab2746
commit 01232e9a1f
4 changed files with 1126 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
freqtrade/freqai/data_drawer.py
View File

@@ -473,7 +473,9 @@ class FreqaiDataDrawer:
model = keras.models.load_model(dk.data_path / f"{dk.model_filename}_model.h5")
elif model_type == 'stable_baselines':
from stable_baselines3.ppo.ppo import PPO
model = PPO.load(dk.data_path / f"{dk.model_filename}_model.zip")
from stable_baselines3 import DQN
#model = PPO.load(dk.data_path / f"{dk.model_filename}_model.zip")
model = DQN.load(dk.data_path / f"best_model.zip")
if Path(dk.data_path / f"{dk.model_filename}_svm_model.joblib").is_file():
dk.svm_model = load(dk.data_path / f"{dk.model_filename}_svm_model.joblib")