Commit Graph

11 Commits

Author SHA1 Message Date
Toshiki Watanabe b30397fff5
Add QR-DQN (#13)
* Add QR-DQN(WIP)

* Update docstring

* Add quantile_huber_loss

* Fix typo

* Remove unnecessary lines

* Update variable names and comments in quantile_huber_loss

* Fix mutable arguments

* Update variable names

* Ignore import not used warnings

* Fix default parameter of optimizer in QR-DQN

* Update quantile_huber_loss to have more reasonable interface

* update tests

* Add assertion to quantile_huber_loss

* Update variable names of quantile regression

* Update comments

* Reduce the number of quantiles during test

* Update comment

* Update quantile_huber_loss

* Fix isort

* Add document of QR-DQN without results

* Update docs

* Fix bugs

* Update doc

* Add comments about shape

* Minor edits

* Update comments

* Add benchmark

* Doc fixes

* Update doc

* Bug fix in saving/loading + update tests

Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
2020-12-21 11:17:48 +01:00
Antonin RAFFIN 6bafcf6e88
Add TimeFeatureWrapper (#7)
* Add TimeFeatureWrapper

* Update README

* Address comments
2020-11-13 13:00:56 +02:00
Antonin RAFFIN 017096371c [ci skip] Update README 2020-10-22 16:52:11 +02:00
Antonin RAFFIN e8093965c7 Fix doc build 2020-10-22 14:46:05 +02:00
Antonin RAFFIN 926e488196 Update wording and links 2020-10-17 17:04:00 +02:00
Anssi "Miffyli" Kanervisto 4a1695cf7b Update README 2020-10-15 02:46:29 +03:00
Antonin RAFFIN 7609c87e84 Cleanup TQC 2020-10-12 19:50:08 +02:00
Antonin RAFFIN 672cb256b1 Update script permissions 2020-09-25 12:53:13 +02:00
Antonin RAFFIN 048a512dd1 Update badge 2020-09-25 12:49:01 +02:00
Antonin RAFFIN 0d9f2e229e Add TQC and base scripts 2020-09-25 12:47:45 +02:00
Antonin RAFFIN 8c282a0288
Initial commit 2020-09-20 22:09:57 +02:00