线性回归

线性回归

1.数据集生成

2.损失函数定义

3.随机梯度下降

Softmax

softmax函数如下

def softmax(X):
    X_exp = X.exp()
    partition = X_exp.sum(dim=1, keepdim=True)
    return X_exp / partition  # 这里应用了广播机制

1.定义模型

2.定义损失函数

3.模型训练

4.模型评估