tf keras,取相同型号的平均重量

modelaverage的Python项目详细描述


模型平均值

` model average是一个pip包,它根据[keras模型中的平均权重](https://stackoverflow.com/questions/48212110/average weights in keras models)的启发,生成“相同模型”中模型权重的平均权重。我创建这个pip包是为了使用像“kubernetes”这样的分布式计算环境。

用法

  1. pip install modelaveragegit clone https://github.com/graykode/modelaverage && python setup.py install
  2. 使用函数average(modellist)
    • 模型列表:模型文件名的列表。
    • 返回:平均重量模型

示例

请参见示例

importtensorflowastffrommodelaverageimportaveragemodellist=['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)forwinaveraged_model.get_weights():print(w.shape)

作者

欢迎加入QQ群-->: 979659372 Python中文网_新手群

推荐PyPI第三方库


热门话题
java使用双一元运算符vs函数<Double,Double>   由于存在巨大的休眠,超出了java GC开销限制   java更改Tomcat中Apache文件上载的临时目录   当PDF位于本地驱动器时,javascript无法在IE中显示PDF   java hibernate如何加载瞬态对象?   java如何判断包属于哪个模块?   创建Word文档时java文件已损坏   java如何将Eclipse中的每个开放项目导出为自己的JAR?   java将带有getter和setter的变量添加到现有类中   java高效地发送多封电子邮件   java读/写。具有特殊字符的txt文件   java如何在导出到jar时包含opencv本机库   java Xstream在未完成时停止写入文件   if语句Java:无法检查布尔值是否为null   文本Java:读取txt文件并将其保存在字符串数组中,但不带反斜杠(空格)?   java如何使用正则表达式替换字符串的一部分   通过Java远程运行Powershell脚本   filenames带有xml文件空指针异常的Java文件uri