update examples

KohlerHECTOR · Aug 10, 2024 · 793fab8 · 793fab8
1 parent c059f00
commit 793fab8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -56,7 +56,7 @@ learner = ObliqueDTPolicy(clf, env)  #
 
 # Start the imitation learning
 interpret = Interpreter(oracle, learner, env)
-interpret.fit(10)
+interpret.fit(5e4)
 
 # Eval and save the best tree
 final_tree_reward, _ = evaluate_policy(interpret._policy, env=env, n_eval_episodes=10)

diff --git a/docs/usage.md b/docs/usage.md
@@ -41,7 +41,7 @@ learner = ObliqueDTPolicy(clf, env)  #
 
 # Start the imitation learning
 interpret = Interpreter(oracle, learner, env)
-interpret.fit(10)
+interpret.fit(5e4)
 
 # Eval and save the best tree
 final_tree_reward, _ = evaluate_policy(interpret._policy, env=env, n_eval_episodes=10)