<template>
<div class="container">
<div class="cal-container">
<div>
计算结果:{{ result }}
</div>
<div>
次数:{{ calculateNum }}
</div>
<button @click="train">计算</button>
<!-- <div class="box"></div> -->
</div>
</div>
</template>
<script>
import * as tf from '@tensorflow/tfjs'
export default {
data() {
return {
result: null,
model: tf.sequential(),
xs: null,
ys:null,
calculateNum:0,
}
},
methods: {
preCalculate() {
this.model.add(tf.layers.dense({units: 1, inputShape: [1]}))
this.model.compile({loss: 'meanSquaredError', optimizer: 'sgd'})
this.xs = tf.tensor2d([1, 2, 3, 4], [4, 1])
this.ys = tf.tensor2d([1, 3, 5, 7], [4, 1])
},
async train(){
await this.preCalculate()
for(let index = 0; index <= 10000; index++){
this.calculateNum = index
await this.model.fit(this.xs, this.ys, {eopchs: 10})
this.result = this.model.predict(tf.tensor2d([5], [1, 1]))
}
}
},
}
</script>
![在这里插入图片描述](https://img-blog.csdnimg.cn/direct/fbe80f35782f4a21aa7c5fb70298567d.png)