Skip to content

Commit

Permalink
docs: duplicate documentation for scheduler_params
Browse files Browse the repository at this point in the history
  • Loading branch information
fortin-alex authored and Hartorn committed Feb 27, 2020
1 parent 6f7c0e0 commit ae49f69
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -120,7 +120,7 @@ You can also get comfortable with how the code works by playing with the **noteb

- scheduler_params : dict

Dictionnary of parameters to apply to the scheduler_fn.
Dictionnary of parameters to apply to the scheduler_fn. Ex : {"gamma": 0.95, "step_size": 10}

- model_name : str (default = 'DreamQuarkTabNet')

Expand All @@ -129,9 +129,7 @@ You can also get comfortable with how the code works by playing with the **noteb
- saving_path : str (default = './')

Path defining where to save models.
- scheduler_params: dict

Parameters dictionnary for the scheduler_fn. Ex : {"gamma": 0.95, "step_size": 10}
- verbose : int (default=1)

Verbosity for notebooks plots, set to 1 to see every epoch, 0 to get None.
Expand Down

0 comments on commit ae49f69

Please sign in to comment.