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

Add total profit to the monitor file and add monitor for the trainingf

Browse Source 
This can help for beginners to see what is happening inside of the training, and helps them understand how much profit of their agent did under the period of training and eval.
This commit is contained in:
richardjozsa 2022-11-15 10:43:28 +01:00
parent 926f870fb5
commit ad56c45ecd
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
freqtrade/freqai/RL/BaseReinforcementLearningModel.py
View File

@ -140,18 +140,18 @@ class BaseReinforcementLearningModel(IFreqaiModel):
train_df = data_dictionary["train_features"]
test_df = data_dictionary["test_features"]
self.train_env = self.MyRLEnv(df=train_df,
self.train_env = Monitor(self.MyRLEnv(df=train_df,
prices=prices_train,
window_size=self.CONV_WIDTH,
reward_kwargs=self.reward_params,
config=self.config,
dp=self.data_provider)
dp=self.data_provider), info_keywords=('total_profit'))
self.eval_env = Monitor(self.MyRLEnv(df=test_df,
prices=prices_test,
window_size=self.CONV_WIDTH,
reward_kwargs=self.reward_params,
config=self.config,
dp=self.data_provider))
dp=self.data_provider), info_keywords=('total_profit'))
self.eval_callback = EvalCallback(self.eval_env, deterministic=True,
render=False, eval_freq=len(train_df),
best_model_save_path=str(dk.data_path))