实现 Tensorflow 和 SaaK Transform 的教程:手写数字识别

Implementing a tutorial for Tensorflow and SaaK Transform: Handwritten Digits Recognition

我正在学习使用 Tensorflow。我已经安装了大部分库。

# load libs
import torch
import argparse
from torchvision import datasets, transforms
import matplotlib.pyplot as plt
import numpy as np
from data.datasets import MNIST
import torch.utils.data as data_utils
from sklearn.decomposition import PCA
import torch.nn.functional as F
from torch.autograd import Variable

但是我遇到了这样的错误:

ImportError Traceback (most recent call last)
<ipython-input-8-257b7af364e1> in <module>()
  5 import matplotlib.pyplot as plt
  6 import numpy as np
----> 7 from data.datasets import MNIST
  8 import torch.utils.data as data_utils
  9 from sklearn.decomposition import PCA

ImportError: No module named data.datasets

我已经使用以下命令安装了 "dataset" 库:

sudo -H pip install dataset

但我仍然看到这个错误。我正在使用 python 2.7。 我对 python 很陌生。任何人都可以帮助我识别丢失的东西。提前致谢。

dataset 目录位于 main.py 旁边,所以当您尝试 sudo -H pip install dataset 时,您可能实际上安装了一个不相关的软件包。

所以卸载错误的包就可以了:

sudo -H pip uninstall dataset