apple / pfl-research

Simulation framework for accelerating research in Private Federated Learning
http://apple.github.io/pfl-research/
Apache License 2.0
281 stars 27 forks source link

Visualize Histogram metrics in Wandb #92

Open grananqvist opened 1 month ago

grananqvist commented 1 month ago

pfl supports reporting metrics to Weights & biases with https://apple.github.io/pfl-research/reference/callback.html#pfl.callback.WandbCallback. Currently we only report the line metrics printed out in console. Wandb has support for visualizing histograms. It can be very useful if we can make use of this feature to visualize histogram of a metric over the cohort of users. This depends on having https://github.com/apple/pfl-research/issues/91 implemented such that we can use that data for wandb visualization.