Skip to content
This repository has been archived by the owner on Jul 22, 2024. It is now read-only.
/ KL-fusion Public archive

This repository contains code for our ICML 2020 paper "Model Fusion with Kullback–Leibler Divergence", S. Claici, M. Yurochkin, S. Ghosh and J. Solomon.

License

Notifications You must be signed in to change notification settings

IBM/KL-fusion

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Model Fusion with Kullback-Leibler Divergence

This repository contains code for our ICML 2020 paper "Model Fusion with Kullback–Leibler Divergence", S. Claici, M. Yurochkin, S. Ghosh and J. Solomon.

Dependencies

The code is written in Python3 and depends on cvxpy, numpy, sklearn, scipy, POT, and matplotlib for plotting.

Running

run_simulation_with_plots.py demonstrates how to use KL-fusion on simulated data and reproduces Figures 6 and 7.

About

This repository contains code for our ICML 2020 paper "Model Fusion with Kullback–Leibler Divergence", S. Claici, M. Yurochkin, S. Ghosh and J. Solomon.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages