Commit Graph

41 Commits

Author SHA1 Message Date
Chintan Shah 7ba7fa320d Using pytorch image 2019-09-07 17:53:46 -04:00
Chintan Shah 69d6c0e053 Add Dockerfile 2019-09-07 17:18:31 -04:00
Yaguang 8c20ca1a9c Adds baseline methods for evaluation. 2019-06-18 13:00:24 -07:00
Yaguang 37f0734cb4
Update README.md 2019-03-18 21:32:57 -07:00
Yaguang 3caf887484 Add data example. 2019-03-17 11:56:55 -07:00
Yaguang 4344dddc6d Add instructions on using HDF5 with python. 2019-03-17 11:43:30 -07:00
Yaguang 81b4626193 Remove unused code. 2019-01-18 19:00:34 -08:00
Yaguang 734ecbc138
Update README.md 2019-01-12 12:04:28 -08:00
Yaguang 763eb7af69 Add pretrained model on PEMS-BAY. 2019-01-10 17:44:07 -08:00
Yaguang ad36deb794
Update README.md
Add more instructions.
2019-01-08 11:33:20 -08:00
liyaguang c8a676604b Add PEMS-BAY configuration. 2019-01-08 11:21:22 -08:00
Yaguang 1481b24b1b
Update README.md
Fix typo in the Graph Construction script (thanks to Zifeng Kang).
2018-12-24 00:47:17 -08:00
Yaguang ac5321fcf7
Update README.md
Add an additional link for data sharing.
2018-11-03 11:54:22 -07:00
Yaguang a8aa1732a9
Update README.md 2018-10-12 13:30:53 -07:00
Yaguang d59d44e4f0 Code refactor. 2018-10-01 17:45:46 -07:00
Yaguang bbc06b6c0c Add log level support and load_dataset method. 2018-10-01 17:45:28 -07:00
Yaguang 88d9fc86d1 Update instructions for graph generation. 2018-10-01 10:54:56 -07:00
Yaguang e6645df191 Update README. 2018-10-01 09:59:47 -07:00
Yaguang 2e4b8c868f
Merge pull request #9 from liyaguang/v2
Upgrading to DCRNN v2.
2018-10-01 09:56:22 -07:00
Yaguang 9520e6cf85 Update pretrained model. 2018-10-01 09:47:23 -07:00
Yaguang e0212cc178 Update README and requirements. 2018-09-30 22:15:27 -07:00
Yaguang 7e05414d99 Combine Val and Test model. 2018-09-30 22:14:45 -07:00
Yaguang 5dc36fed7c Refactor DCRNN Model. 2018-09-30 21:53:40 -07:00
Yaguang 2f2d748b45 Add AMSGrad from https://github.com/taki0112/AMSGrad-Tensorflow for stablized Adam Training. 2018-09-30 21:52:20 -07:00
Yaguang e8d6c08db0 Merge log_helper into utils and change logging mechanism. 2018-09-26 11:33:21 -07:00
Yaguang 9ec161543b Code refactoring, including data loading, logging, configuration, removing redundant code. 2018-09-26 11:19:00 -07:00
Yaguang 80e156c580
Update README.md
Add data file name.
2018-07-22 18:55:35 +08:00
Yaguang 67bd763b18
Update README.md
Add data file name.
2018-07-22 18:53:50 +08:00
Yaguang 17927239a7 Update the pretrained model and results. 2018-06-07 11:18:55 +08:00
Yaguang e08002d72d Update the seq2seq training logic. 2018-06-07 11:16:33 +08:00
Yaguang 562514cc30 Add a more data-efficient training data generation method. 2018-06-07 10:41:47 +08:00
Yaguang e93435c598 Replace json with yaml for configuration. 2018-04-18 11:51:35 -07:00
Yaguang 3e94a0ff0e Adds support for python 3.5 and python 3.6. 2018-04-18 11:50:39 -07:00
Yaguang fcdf62d6de Update reference. 2018-03-30 13:42:12 -07:00
Yaguang 90e39f5a4a Update README to include the lat/lon file. 2018-03-29 17:58:52 -07:00
Yaguang 82922c8308 Adds the file containing sensor lat/lon. 2018-03-29 17:48:51 -07:00
Yaguang 94b831fc12 Update dependence and change script name. 2018-03-29 15:29:46 -07:00
Yaguang 1063bb995a Update dependence. 2018-03-22 09:42:36 -07:00
Yaguang f81e027489 Adds missing file for generating sensor graph. 2018-03-21 10:43:48 -07:00
Yaguang dad3287945 Update citation information. 2018-02-25 20:44:55 -08:00
Yaguang c7535e5443 Initial commit. 2017-12-07 17:35:50 -08:00