modelaverage 1.0.1

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

modelaverage

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

Usage

1. pip install modelaverage or git clone https://github.com/graykode/modelaverage && python setup.py install
2. using function average(modellist)
   • modellist : list of model file names.
   • return : averaged weight model

Example

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():
    print(w.shape)

Author

• Name : Tae Hwan Jung(@graykode)
• Email : [e-mail]