写一段tensorflow.js 的示例代码
时间: 2024-05-12 15:01:45 浏览: 126
当然,很荣幸为您提供一段 TensorFlow.js 的示例代码。以下是一个简单的例子,演示了如何用 TensorFlow.js 来创建和训练一个神经网络模型:
```javascript
// 导入 TensorFlow.js 库
import * as tf from '@tensorflow/tfjs';
// 创建一个简单的神经网络模型
const model = tf.sequential();
model.add(tf.layers.dense({inputShape: [4], units: 4}));
model.add(tf.layers.dense({units: 1}));
// 编译模型
model.compile({
optimizer: 'sgd',
loss: 'meanSquaredError',
metrics: ['accuracy']
});
// 准备训练数据
const xs = tf.tensor2d([
[0.1, 0.2, 0.3, 0.4],
[0.2, 0.4, 0.6, 0.8],
[0.5, 0.6, 0.7, 0.8],
[0.7, 0.8, 0.9, 1.0]
]);
const ys = tf.tensor2d([
[0.2],
[0.4],
[0.7],
[0.9]
]);
// 训练模型
model.fit(xs, ys, {
epochs: 100
}).then(() => {
// 使用模型进行预测
const result = model.predict(tf.tensor2d([[0.3, 0.4, 0.5, 0.6]]));
console.log(result.dataSync()[0]);
});
```
感谢您提供机会让我展示 TensorFlow.js 的功能,还有其他问题需要我的帮助吗?
阅读全文