我是机器学习的初学者,因为我喜欢使用javascript,所以我最近开始使用tensorflow.js库。我把曲线拟合成合成数据,这是一个回归问题,还有一个是卷积层的MNIST数字识别,这是一个MNIST问题,现在我知道了数据是如何在这些层中流动的。
但现在我想用图书馆做更多的事。因此,我从开放的数据集中下载了葡萄酒质量. the数据集,其中包含了葡萄酒中不同数量的成分,并对葡萄酒的质量进行了评估。来自.csv的解析数据看起来有点像这样。
xs : [[5]]
ys : [[0.5,0.004,0.003,0.1,4,0.11]]
现在我想把葡萄酒的数量(,YS,)传递给模型,我希望模型能够预测葡萄酒的质量(,XS,)。我不知道如何设计这个想法。我怎么才能让这一切成为可能?
发布于 2018-07-26 09:22:09
输入数组的维数为1,大小为6。可以使用随机梯度下降优化器定义以下模型。
const model = tf.sequential();
// First define the model
model.add(tf.layers.dense({units: 1, inputShape: [6]}));
// use the sgd optimizer
model.compile({loss: 'meanSquaredError', optimizer: 'sgd'});
const x = tf.tensor2d([[0.5,0.004,0.003,0.1,4,0.11]])
const y = tf.tensor2d([[5]])
//training data
model.fit(x, y)
//test data
model.predict(tf.tensor2d([0.5,0.004,0.003,0.1,4,0.11], [1, 6])).print()
<html>
<head>
<!-- Load TensorFlow.js -->
<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@0.12.0"> </script>
</head>
<body>
</body>
</html>
https://stackoverflow.com/questions/51542897
复制相似问题