/ml-torch

pytorch实现常见的机器学习算法

Primary LanguageJupyter Notebook

ml-torch

机器学习算法实现

  • 该项目中实现的算法主要包括逻辑回归、softmax回归、全连接前馈神经网络、卷积神经网络
  • 每个算法都包含两种实现方法
    • 简单实现:利用pytorch提供的torch.nn库来实现
    • 复杂实现:利用pytorch中的张量,从零开始实现
  • 逻辑回归的数据集已经给出,其他算法所使用的数据集均为torchvision中的FashionMnist数据集
  • 算法的实现细节解析请参考我的博客