automate eval freq in multiproc
This commit is contained in:
parent
94cfc8e63f
commit
05ccebf9a1
@ -62,7 +62,6 @@ class ReinforcementLearner_multiproc(BaseReinforcementLearningModel):
|
|||||||
"""
|
"""
|
||||||
train_df = data_dictionary["train_features"]
|
train_df = data_dictionary["train_features"]
|
||||||
test_df = data_dictionary["test_features"]
|
test_df = data_dictionary["test_features"]
|
||||||
eval_freq = self.freqai_info["rl_config"]["eval_cycles"] * len(test_df)
|
|
||||||
|
|
||||||
env_id = "train_env"
|
env_id = "train_env"
|
||||||
num_cpu = int(self.freqai_info["rl_config"]["thread_count"] / 2)
|
num_cpu = int(self.freqai_info["rl_config"]["thread_count"] / 2)
|
||||||
@ -77,5 +76,5 @@ class ReinforcementLearner_multiproc(BaseReinforcementLearningModel):
|
|||||||
config=self.config) for i
|
config=self.config) for i
|
||||||
in range(num_cpu)])
|
in range(num_cpu)])
|
||||||
self.eval_callback = EvalCallback(self.eval_env, deterministic=True,
|
self.eval_callback = EvalCallback(self.eval_env, deterministic=True,
|
||||||
render=False, eval_freq=eval_freq,
|
render=False, eval_freq=len(train_df),
|
||||||
best_model_save_path=dk.data_path)
|
best_model_save_path=dk.data_path)
|
||||||
|
Loading…
Reference in New Issue
Block a user