基于TypeScript的机器学习项目
https://github.com/ounana/infers github慢的话码云有同名镜像库 示例地址 矩阵运算,加、减、乘、除、转置、行列式 简单的卷积网络CNN,基于BP反向传播算法 可用的激活函数:‘Sigmoid’ | ‘Relu’ | ‘Tanh’ | 'Softmax’ 三种训练模式:bgd | sgd | mbgd
XOR异或例子,这是三层的神经网络
let xs = new Matrix([[1, 0], [0, 1], [0, 0], [1, 1]])
let ys = new Matrix([[1], [1], [0], [0]])
let model = new BPNet([2, [6, 'Tanh'], [1, 'Sigmoid']], { rate: 0.1 })
model.fit(xs, ys, {
epochs: 5000, onEpoch: (epoch, loss) => {
if (epoch % 100 === 0) console.log('epoch = ' + epoch, loss)
}
})
model.predict(xs).print()
// Matrix 4x1 [
// 0.9862025352830867,
// 0.986128496195502,
// 0.01443800549676924,
// 0.014425871504885788,
// ]
1 回复
更新了示例