JS Tensorflow 首次尝试

最新推荐文章于 2023-01-17 15:10:23 发布

天问

最新推荐文章于 2023-01-17 15:10:23 发布

阅读量328

点赞数

分类专栏： tensorflow 文章标签： js tensorflow

本文链接：https://blog.csdn.net/STK_tianwen/article/details/81556703

版权

tensorflow 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

本文通过一个简单的线性回归实例介绍了如何使用TensorFlow.js进行模型训练，并展示了从数据生成到模型预测的全过程。作者在尝试使用TensorFlow.js实现Y=2*X-1这一简单线性关系时遇到了预测结果偏离预期的问题。

摘要由CSDN通过智能技术生成

试了一下 tensorflow 的 js版本，就使用的官网的示例代码，感觉结果差很多啊。

目标是 Y = 2*X -1；但结果感觉相差十万八千里。

代码：

<!DOCTYPE html>
<html>
<head>
    <title>JS Tensorflow</title>
    <!-- <script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@latest"></script> -->
    <script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@0.12.0"></script>
    <!-- <script src="https://cdn.bootcss.com/tensorflow/0.11.2/tf.min.js"></script> -->
</head>
<body>
<div id="output"></div>
</body>
<script type="text/javascript">
    // Define a model for linear regression.
const model = tf.sequential();
model.add(tf.layers.dense({units: 1, inputShape: [1]}));

// Prepare the model for training: Specify the loss and the optimizer.
model.compile({loss: 'meanSquaredError', optimizer: 'sgd'});

// Generate some synthetic data for training.
// y = 2*X-1
var x = [], y=[], len = 20;
for (var i = 0; i < len; i++) {
    let v = Math.floor(Math.random()*100)-50;
    x.push(v);
    y.push((v<<1)-1);
}
// console.log(x);
// console.log(y);
const xs = tf.tensor2d(x, [len, 1]);
// xs.print();
const ys = tf.tensor2d(y, [len, 1]);
// ys.print();

// Train the model using the data.
model.fit(xs, ys).then(() => {
  // Use the model to do inference on a data point the model hasn't seen before:
  // Open the browser devtools to see the output
  model.predict(tf.tensor2d([10], [1, 1])).print();
  // document.getElementById('output').innerHTML = model.predict( tf.tensor2d([10], [1, 1]) );
});
      
</script>
</html>

结果：

Tensor
[[281.0496216],]

这差了太多了。这是完全不懂，刚开始学习，照例子写了一下，慢慢学习看看哪里有问题