维科纳蒂 »
×
改变方向
在 Codepen 上打开
更改主题,深色/浅色
<!DOCTYPE html> <html> <script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs"></script> <script src="https://cdn.plot.ly/plotly-latest.min.js"></script> <body> <h2>TensorFlow.js</h2> <p id="message">Модель тренується!</p> <div id="myPlot" style="width:100%;max-width:700px"></div> <script> // Створення навчальних даних const xs = tf.tensor([0, 1, 2, 3, 4]); const ys = xs.mul(1.2).add(5); // Визначте модель лінійної регресії const model = tf.sequential(); model.add(tf.layers.dense({units:1, inputShape:[1]})); // Вкажіть Loss і Optimizer model.compile({loss: 'meanSquaredError', optimizer:'sgd'}); // Навчіть модель model.fit(xs, ys, {epochs:500}).then(() => {myFunction()}); // Використовуйте модель function myFunction() { const xMax = 20; const xArr = []; const yArr = []; for (let x = 10; x <= xMax; x++) { let result = model.predict(tf.tensor([Number(x)])); result.data().then(y => { xArr.push(x); yArr.push(Number(y)); if (x == xMax) {display(xArr,yArr)}; }); } } function display(xArr, yArr) { let text = "Правильно передбачено<br>"; for (let i = 0; i < xArr.length; i++) { text += (xArr[i]*1.2+5).toFixed(4) + " " + yArr[i].toFixed(4) + "<br>"; } document.getElementById('message').innerHTML = text; } </script> </body> </html>