|
223 | 223 | " self,\n", |
224 | 224 | " add_exit_choice=False, # Whether to add exit choice with utility=1\n", |
225 | 225 | " optimizer=\"lbfgs\", # Optimizer to use\n", |
226 | | - " tolerance=1e-8, # Absolute function tolerance for optimality if lbfgs is used\n", |
| 226 | + " lbfgs_tolerance=1e-8, # Absolute function tolerance for optimality if lbfgs is used\n", |
227 | 227 | " lr=0.001, # learning rate if stochastic gradient descent optimizer\n", |
228 | 228 | " epochs=1000, # maximum number of epochs\n", |
229 | 229 | " batch_size=32, # batch size if stochastic gradient descent optimizer\n", |
230 | 230 | " ):\n", |
231 | 231 | " \"\"\"Model coefficients instantiation.\"\"\"\n", |
232 | 232 | " super().__init__(add_exit_choice=add_exit_choice,\n", |
233 | 233 | " optimizer=optimizer,\n", |
234 | | - " tolerance=tolerance,\n", |
| 234 | + " lbfgs_tolerance=lbfgs_tolerance,\n", |
235 | 235 | " lr=lr,\n", |
236 | 236 | " epochs=epochs,\n", |
237 | 237 | " batch_size=batch_size)\n", |
|
592 | 592 | ], |
593 | 593 | "metadata": { |
594 | 594 | "kernelspec": { |
595 | | - "display_name": "tf_env", |
| 595 | + "display_name": "base", |
596 | 596 | "language": "python", |
597 | 597 | "name": "python3" |
598 | 598 | }, |
|
0 commit comments