定义模型结构-可视化训练模型过程day03

定义模型结构:单层单个神经元组成的神经网络

操作步骤:

  1. 初始化一个神经网络模型
  2. 为神经网络模型添加层
  3. 设计层的神经元个数和inputShape(输入形状)
    // 连续的模型
    const model = tf.sequential(); // 全链阶层 /* dense:全链阶层 units:神经元个数1,单个足够用 inputShape: 1维 */ model.add(tf.layers.dense({ units:1, inputShape: [1] }));

损失函数,均方误差MSE

操作步骤:

  1. 跟着Google机器学习速成教程理解损失函数与均方误差;
    (机器学习速成教程)https://developers.google.cn/machine-learning/crash-course/descending-into-ml/training-and-loss

  2. 在Tensorflow.js中设置损失函数

    // 损失函数
    /*
    meanSquaredError均方误差
    */
    model.compile({ loss: tf.losses.meanSquaredError });

优化器:随机剃度下降SGD

操作步骤:
model.compile({ loss: tf.losses.meanSquaredError, optimizer: tf.train.sgd() });

训练模型并可视化训练过程

  1. 将训练数据转为Tensor
  2. 训练类型
  3. 使用tfvis可视化训练过程
    配置package.json(适配async await)
    "browserslist": ["last 1 Chrome version"]

` const inputs = tf.tensor(xs); // 将数据转为tensor
const labels = tf.tensor(ys);
await model.fit(inputs, labels, {
batchSize: 1, // 样本
epochs: 100, // 训练几次
callbacks: tfvis.show.fitCallbacks(
{ name: '训练过程' },
['loss']

    )
});`


image

posted @   小白咚  阅读(129)  评论(0编辑  收藏  举报
编辑推荐:
· AI与.NET技术实操系列(二):开始使用ML.NET
· 记一次.NET内存居高不下排查解决与启示
· 探究高空视频全景AR技术的实现原理
· 理解Rust引用及其生命周期标识(上)
· 浏览器原生「磁吸」效果!Anchor Positioning 锚点定位神器解析
阅读排行:
· DeepSeek 开源周回顾「GitHub 热点速览」
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET 10首个预览版发布:重大改进与新特性概览!
· AI与.NET技术实操系列(二):开始使用ML.NET
· 单线程的Redis速度为什么快?
点击右上角即可分享
微信分享提示