数据集以及源码在这里获取:
https://github.com/beyondguo/LeaningDeepLearning
import numpy as np
import h5py #一个h5py文件是 “dataset” 和 “group” 二合一的容器。
def load_dataset():
train_dataset=h5py.File('/Users/momozi1996/Downloads/LeaningDeepLearning-master/LR/datasets/train_catvnoncat.h5',"r") #读取h5格式的图片
train_set_x_orig=np.array(train_dataset["train_set_x"][:]) #train set features 创建了数组
train_set_y_orig=np.array(train_dataset["train_set_y"][:]) #train set label
test_dataset=h5py.File('/Users/momozi1996/Downloads/LeaningDeepLearning-master/LR/datasets/test_catvnoncat.h5',"r") #读取h5格式的图片
test_set_x_orig=np.array(test_dataset["test_set_x"][:]) #test set features 创建了数组
test_set_y_orig=np.array(test_dataset["test_set_y"][:]) #test set label
classes=np.array(test_dataset["list_classes"][:]) # the list of classes
train_set_y_orig=train_set_y_ori