当然可以
我来写一个例子
<script type="text/javascript" src="http://repo.bfw.wiki/bfwrepo/js/tf.min.js"></script>
<script type="text/javascript">
async function predictfuture() {
////////////////////////
// create fake data
///////////////////////
var xs = tf.tensor3d([
[[1], [1], [0]],
[[1], [1], [0]],
[[1], [1], [0]],
[[1], [1], [0]],
[[1], [1], [0]],
[[1], [1], [0]]
]);
xs.print();
var ys = tf.tensor3d([
[[1], [1], [0]],
[[1], [1], [0]],
[[1], [1], [0]],
[[1], [1], [0]],
[[1], [1], [0]],
[[1], [1], [0]]
]);
ys.print();
////////////////////////
// create model w/ layers api
///////////////////////
console.log('Creating Model...');
/*
model design:
i(xs) h o(ys)
batch_size -> * * * -> batch_size
timesteps -> * * * -> timesteps
input_dim -> * * * -> input_dim
*/
const model = tf.sequential();
const hidden = tf.layers.lstm({
units: 3,
activation: 'sigmoid',
inputShape: [3, 1],
returnSequences: true
});
model.add(hidden);
//output layer
const output = tf.layers.lstm({
units: 1,
activation: 'sigmoid',
returnSequences: true
})
model.add(output);
//compile
const sgdoptimizer = tf.train.sgd(0.1)
model.compile({
optimizer: sgdoptimizer,
loss: tf.losses.meanSquaredError
});
////////////////////////
// train & predict
///////////////////////
console.log('Training Model...');
await model.fit(xs, ys, {
epochs: 200
}).then(() => {
console.log('Training Complete!');
console.log('Creating Prediction...');
const inputs = tf.tensor3d( [[[1],[1],[0]]] );
// const inputs = tf.tensor2d([[1], [1], [0]]);
let outputs = model.predict(inputs);
outputs.print();
});
}
predictfuture();
</script>
<h2>tensorflow.js预测未来</h2>
请在console总查看效果
网友回复
腾讯混元模型广场里都是混元模型的垂直小模型,如何api调用?
为啥所有的照片分辨率提升工具都会修改照片上的图案细节?
js如何在浏览器中将webm视频的声音分离为单独音频?
微信小程序如何播放第三方域名url的mp4视频?
ai多模态大模型能实时识别视频中的手语为文字吗?
如何远程调试别人的chrome浏览器获取调试信息?
为啥js打开新网页window.open设置窗口宽高无效?
浏览器中js的navigator.mediaDevices.getDisplayMedia屏幕录像无法录制SpeechSynthesisUtterance产生的说话声音?
js中mediaRecorder如何录制window.speechSynthesis声音音频并下载?
python如何直接获取抖音短视频的音频文件url?