吴恩达机器学习作业Python实现之神经网络

import os

os.chdir('E:/ML/machine-learning-ex4')
print('现在工作目录是 '+str(os.getcwd()))



import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
import scipy.io as sio

data=sio.loadmat('ex4data1.mat')

X=data['X'];y=data['y']

# 可视化X中部分数据
fig,axs=plt.subplots(3,3)
axs=axs[:,:].flatten()
import random
for i in range(len(axs)):
    j=random.randint(0,len(X)-1)
    num=int(np.sqrt(len(X[0,:])))
    axs[i].imshow(X[j,:].reshape(num,num).T)
    axs[i].set_ylabel(str(y[j]))
  
    
plt.show()