mohammadian7/freqtrade
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use Cloudpickle also for reading

Browse Source
This commit is contained in:
Matthias
2023-11-25 19:37:49 +01:00
parent eeb460e55c
commit 72eeb6561c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
freqtrade/freqai/data_drawer.py
+2 -2
View File
@@ -12,7 +12,6 @@ import numpy as np
import pandas as pd import pandas as pd
import psutil import psutil
import rapidjson import rapidjson
from joblib import load
from joblib.externals import cloudpickle from joblib.externals import cloudpickle
from numpy.typing import NDArray from numpy.typing import NDArray
from pandas import DataFrame from pandas import DataFrame
@@ -559,7 +558,8 @@ class FreqaiDataDrawer:
if dk.live and coin in self.model_dictionary: if dk.live and coin in self.model_dictionary:
model = self.model_dictionary[coin] model = self.model_dictionary[coin]
elif self.model_type == 'joblib': elif self.model_type == 'joblib':
model = load(dk.data_path / f"{dk.model_filename}_model.joblib") with (dk.data_path / f"{dk.model_filename}_model.joblib").open("rb") as fp:
model = cloudpickle.load(fp)
elif 'stable_baselines' in self.model_type or 'sb3_contrib' == self.model_type: elif 'stable_baselines' in self.model_type or 'sb3_contrib' == self.model_type:
mod = importlib.import_module( mod = importlib.import_module(
self.model_type, self.freqai_info['rl_config']['model_type']) self.model_type, self.freqai_info['rl_config']['model_type'])