IBM / federated-learning-lib

A library for federated learning (a distributed machine learning process) in an enterprise environment.
Other
500 stars 136 forks source link

Update generate_configs.py #132

Open mzamini92 opened 1 year ago

mzamini92 commented 1 year ago

some improvements on generate_configs file. Instead of directly comparing the dataset names in get_data_handler_config, we can use constants to make the code more readable and maintainable. Rather than passing strings for model types like "sklearn" we used enums to represent different model types. Improved the serialization technique. Signed-off-by: Mohamad Zamini mohamad.zamini@gmail.com