CNode

基于TypeScript的机器学习项目

分享
Oounana发布于5 年前最后回复5 年前1 回复4473 浏览0 收藏

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)

O
ounana#1·5 年前

更新了示例

参与回复
登录后即可参与回复。登录