├── .gitignore ├── LICENSE ├── README.md ├── arg_parser.py ├── client_dataset.py ├── client_trainer.py ├── constants.py ├── data └── MIT Stanford battery dataset │ ├── .gitkeep │ └── preprocessed_data.pkl ├── environment.yml ├── fl_platform_overview.png ├── masked_mae.py ├── masked_mape_loss.py ├── masked_rmse_loss.py ├── mit_stanford_data_preprocessing.ipynb ├── mit_stanford_features_dataset.py ├── results └── .gitkeep ├── rmse_loss.py ├── server_trainer.py └── train_fl_lstm.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KeiLongW/decentralized-battery-health-forecast/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KeiLongW/decentralized-battery-health-forecast/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KeiLongW/decentralized-battery-health-forecast/HEAD/README.md -------------------------------------------------------------------------------- /arg_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KeiLongW/decentralized-battery-health-forecast/HEAD/arg_parser.py -------------------------------------------------------------------------------- /client_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KeiLongW/decentralized-battery-health-forecast/HEAD/client_dataset.py -------------------------------------------------------------------------------- /client_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KeiLongW/decentralized-battery-health-forecast/HEAD/client_trainer.py -------------------------------------------------------------------------------- /constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KeiLongW/decentralized-battery-health-forecast/HEAD/constants.py -------------------------------------------------------------------------------- /data/MIT Stanford battery dataset/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/MIT Stanford battery dataset/preprocessed_data.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KeiLongW/decentralized-battery-health-forecast/HEAD/data/MIT Stanford battery dataset/preprocessed_data.pkl -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KeiLongW/decentralized-battery-health-forecast/HEAD/environment.yml -------------------------------------------------------------------------------- /fl_platform_overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KeiLongW/decentralized-battery-health-forecast/HEAD/fl_platform_overview.png -------------------------------------------------------------------------------- /masked_mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KeiLongW/decentralized-battery-health-forecast/HEAD/masked_mae.py -------------------------------------------------------------------------------- /masked_mape_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KeiLongW/decentralized-battery-health-forecast/HEAD/masked_mape_loss.py -------------------------------------------------------------------------------- /masked_rmse_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KeiLongW/decentralized-battery-health-forecast/HEAD/masked_rmse_loss.py -------------------------------------------------------------------------------- /mit_stanford_data_preprocessing.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KeiLongW/decentralized-battery-health-forecast/HEAD/mit_stanford_data_preprocessing.ipynb -------------------------------------------------------------------------------- /mit_stanford_features_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KeiLongW/decentralized-battery-health-forecast/HEAD/mit_stanford_features_dataset.py -------------------------------------------------------------------------------- /results/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /rmse_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KeiLongW/decentralized-battery-health-forecast/HEAD/rmse_loss.py -------------------------------------------------------------------------------- /server_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KeiLongW/decentralized-battery-health-forecast/HEAD/server_trainer.py -------------------------------------------------------------------------------- /train_fl_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/KeiLongW/decentralized-battery-health-forecast/HEAD/train_fl_lstm.py --------------------------------------------------------------------------------