Codes of FedSampling
- Utils.py: Codes of data preprocessing functions
- DataLoader.py: Codes for loading and preprocessing text and images data
- Partation.py: Codes for class non-IID and size non-IID data partition
- Models.py: Codes for implementing the basic text and image models
- FedSampling.py: Codes of the proposed federated data sampling and training mechanism
- Main-SizeNonIID.ipynb: Codes of conducting the size non-IID experiments
- Main-ClassNonIID.ipynb: Codes of conducting the class non-IID experiments