Skip to content

Cardio-AI/federated-dashboard

Repository files navigation

Federated Learning Dashboard

This dashboard provides a live overview of a federated training scenario. The average metric or loss curve for all nodes is provided as well as general information about the latest training run, the nodes as well as data used, and predictions of the remaining epoch and training duration. It visualises the metric and loss curves for every node as well as node connectivity states and error logging.

The Federated Learning Dashboard

Installation and setup

Run using Docker

Just run docker compose up. This will create the three containers (server, DB, UI), start the UI on http://localhost:3000 and makes the server accessible on http://localhost:5000.

Install the package

The package can be installed using:

pip install git+https://github.com/Cardio-AI/federated-dashboard.git

for example usage please refer to this example script.

Structure

The repository consists of three parts:

  • Python package: This package provides an intuitive API which allows logging of events and statuses from any federated training setup to abstract the communication to the server.
  • Backend server with postgres database: This part provides routes for the UI and the package.
  • React client UI: This browser application visualises the data provided by the server.

This project is heavily inspired by Tensorboard and features a nearly identical API for easily integrated event logging.

It is currently in development (pull requests are welcome) but already features convenience features such as automatic updates and automatic theme detection.

Light Mode

About

A dashboard to visualise federated training

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published