Skip to main content

tf-keras, make the average of model weight in same models

Project description


`modelaverage` is a pip package which make the average weight of model weight in `same models`, inspired by [Average weights in keras models]( I created this pip package to use distributed computing environment like `kubernetes`.


  1. pip install modelaverage or git clone && python install
  2. using function average(modellist)
    • modellist : list of model file names.
    • return : averaged weight model


Please see example

import tensorflow as tf
from modelaverage import average

modellist = ['models/mnist1.h5', 'models/mnist2.h5', 'models/mnist3.h5', 'models/mnist4.h5', 'models/mnist5.h5',
             'models/mnist6.h5', 'models/mnist7.h5', 'models/mnist8.h5', 'models/mnist9.h5']

averaged_model = average(modellist)

for w in averaged_model.get_weights():


Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Filename, size & hash SHA256 hash help File type Python version Upload date
modelaverage-1.0.1-py3.6.egg (3.8 kB) Copy SHA256 hash SHA256 Egg 3.6
modelaverage-1.0.1.tar.gz (2.1 kB) Copy SHA256 hash SHA256 Source None

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN SignalFx SignalFx Supporter DigiCert DigiCert EV certificate StatusPage StatusPage Status page