基于TypeScript的机器学习项目
发布于 4 年前 作者 ounana 3740 次浏览 来自 分享

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 回复

更新了示例

回到顶部