机器学习
COSummer
生物医学小硕一枚,现从事医疗软件研发
展开
-
mlpack 训练随机森林分类模型,并保存、加载网络,预测结果
mlpack 是一个C++的机器学习库,它重点在于其扩展性、高速性和易用性。它的目的是让新用户通过简单、一致的API使用机器学习,同时为专业用户提供C++的高性能和最大灵活性。他的性能超出大量类似的机器学习库,如WEKA、Shogun、MATLAB、mlpy及sklearn以下代码示例了mlpack 训练随机森林分类模型,并保存、加载网络,预测结果#include "mlpack/core.hpp"#include "mlpack/methods/random_forest/random_for原创 2020-11-27 18:19:46 · 2233 阅读 · 1 评论 -
.h5 To Pb
以下代码实现了把.h5文件转换成pb的过程from keras.models import load_modelimport tensorflow as tfimport osimport os.path as ospfrom keras import backend as K#路径参数input_path = '/home/simon/ASTAR Validation/Code...原创 2019-10-18 15:29:05 · 720 阅读 · 0 评论 -
ckpt2pb
import os, argparseimport tensorflow as tf# The original freeze_graph function# from tensorflow.python.tools.freeze_graph import freeze_graph dir = os.path.dirname(os.path.realpath(__file__))...原创 2019-07-01 09:51:05 · 367 阅读 · 1 评论