机器学习
纳米时速
这个作者很懒,什么都没留下…
展开
-
tensorflow 对GPU的配置
tf1配置import keras.backend.tensorflow_backend as KTFfrom tensorflow import set_random_seedconfig = tf.ConfigProto(allow_soft_placement=True)# 设置最大占有GPU不超过显存的80%config.gpu_options.per_process_gpu_memory_fraction = 0.8# GPU资源使用率,先少量分配,按需逐步增加...原创 2021-04-21 15:00:06 · 467 阅读 · 0 评论 -
机器学习之keras模型保存为pb文件
keras模型是依赖tensorflow框架的,在恢复模型之前还需要再定义一遍网络结构,这对于部署到生产环境来说非常不方便。而转换为pb文件,可以独立运行,任何语言都可以解析它,同时方便部署到tf serving上。本文提供以下两种转换方法。方法1(推荐):适用于tf2.0之后的版本,但是1.0版本生成的hdf5文件也可以用此方法转换,前提必须在tf2.0环境下运行model_path = 'weights.hdf5' # 模型文件model =.原创 2021-04-16 10:08:39 · 3046 阅读 · 0 评论 -
python机器学习常用包
1、Numpy使用Python进行科学计算的基本软件包,是scikit-learn和SciPy等机器学习库的基础。链接:https://numpy.org/安装方式:pippip install numpycondaconda install numpy2、Pandas快速,强大,灵活且易于使用的开源数据分析和处理工具。链接:https://pandas.pydata.org/...原创 2021-04-14 16:13:02 · 4853 阅读 · 1 评论