数据集下载:
train_catvnoncat.h5(训练集),test_catvnoncat.h5(测试集),lr_utils.py
链接: https://pan.baidu.com/s/16o6Wb9r6YSjH8PUdGz9gnw 提取码: zbb6
lr_utils.py是吴恩达老师给定的一个文件,不是python中自带的一个module,所以是不能通过pip命令安装的
lr_utils.py代码如下
import h5py import numpy as np def load_dataset(): train_dataset = h5py.File('datasets/train_catvnoncat.h5', "r") train_set_x_orig = np.array(train_dataset["train_set_x"][:]) # your train set features train_set_y_orig = np.array(train_dataset["train_set_y"][:]) # your train set labels test_dataset = h5py.File('datasets/test_catvnoncat.h5', "r") test_set_x_orig = np.array(test_dataset["test_set_x"][:]) # your test set features test_set_y_orig = np.array(test_dataset["test_set_y"][:]) # your test set labels classes = np.array(test_dataset["list_classes"][:]) # the list of classes train_set_y_orig = train_set_y_orig.reshape((1, train_set_y_orig.shape[0])) test_set_y_orig = test_set_y_orig.reshape((1, test_set_y_orig.shape[0])) return train_set_x_orig, train_set_y_orig, test_set_x_orig, test_set_y_orig, classes
Pycharm数据集放此路径下
jupyter:放在同一文件夹下用jupyter打开ipynb