- 博客(3)
- 收藏
- 关注
原创 4 5 分类代码
将当前批次中每个样本的预测类别索引pred_value转换为 NumPy 数组并进一步转换为 Python 列表,然后将其添加到labels列表中,,这通常用于在推理过程中手机所有样本的预测类别。#遍历pred_prob列表中的每一个元素,并同时获取每个元素的索引和值 enumerate可以在迭代过程返回每个元素的索引和对应的值。#定义了一个类的构造函数(即初始化方法)self指向当前实例的引用 num_class是一个用户自定义的参数,表示分类任务中的类别数量。
2025-03-16 22:19:05
1206
原创 【李哥考研】回归实战
models模型 , trainloader训练数据的加载器 迭代器 , valloader 验证数据加载器,optimizer 优化器 更新模型参数 , loss损失函数 , epoch 训练的轮数 , device 设备 ,表示是否保存模型的标志。# X_new = get_feature_importance_with_pca(x, feature_dim, column) # 选重要的dim列。这个函数的目的是, 找到所有的特征种, 比较有用的k个特征, 并打印这些列的名字。
2025-02-18 13:11:26
502
原创 【李哥考研】线性表示代码代写
生成符合正态分布的随机数 0正态分布的均值,1正态分布的标准差,(data_num, len(w))输出张量的形状(维度和大小)#生成数据 w权重向量,b偏置项,data_num需要生成的数据点数量。print("真实的函数值是",true_w,true_b)#绘制散点图 1指定散点图中点的大小(默认20)#返回当前批次的数据和标签 有存档点的return。print("训练得到的参数值是",w_0,b_0)#从打乱后的索引列表中提取当前批次的索引。#遍历每个批次的数据和标签。
2025-01-17 16:19:04
776
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人