ML模型
许多常见ML模型的自制实现,包括神经网络,线性模型和决策树。
强调
MNIST基准
导入/获取数据
import numpy as np
np . random . seed ( 0 )
import matplotlib . pyplot as plt
% matplotlib inline
from functools import reduce
import neuralNet as nn
import utils as u
from torchvision import datasets # Only to get MNIST data
train_dataset = datasets . MNIST ( './data' , train = True , download = True )
test_dataset = datasets . MNIST ( './da
评论0
最新资源