Tensorflow.js运行Python下训练的模型

最近tensorflowjs更新了,然后这篇的方法似乎有点行不通,如果失效请参考新的博客:Tensorflow.js运行Python下训练的CNN模型

一、引言

这两天的项目需要用到Tensorflow.js来实现一个AI,尽管说Tensorflow.js本身是有训练模型的功能的,不过考虑到javascript这个东西加载资源要考虑跨域问题等种种因素。。最终还是决定使用python的tensorflow来训练模型,然后利用js端来使用模型进行运算,那么关键问题就是:js如何加载python下训练的模型

【webAI】Tensorflow.js加载预训练的model

这位博主的博客给了我很大的帮助,前两步按照他的教程来做都是没有什么问题的,不过其实还是有一些潜在的坑或者对于我这种前端小白不太友好的地方,这里我把我的整个过程都来叙述一遍吧。

注:首先在命令行中执行

pip install tensorflowjs

安装模型转换的部分,否则转换可能会报错

二、python部分

这里我用了一个更加简单的例子,and.py,让神经网络来学习异或运算(忽略这个"and"emmm),这里我直接把python代码贴出来:

#coding=utf-8#
import tensorflow as tf
import numpy as np
x_data=[[0.0,0.0],[0.0,1.0],[1.0,0.0],[1.0,1.0]]	#训练数据
y_data=[[0.0],[1.0],[1.0],[0.0]]	#标签
x_test=[[0.0,1.0],
评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值