Cs231n_Assignment1_Q3.docx
需积分: 5 59 浏览量
2021-07-30
16:50:38
上传
评论
收藏 721KB DOCX 举报
Cs231n_Assignment1_Q3
Softmax
Step1:初始化、加载数据、可视化数据
In[1]:
1. defget_CIFAR10_data(num_training=49000,num_validation=1000,num_
test=1000,num_dev=500):
2. """
3. LoadtheCIFAR-10datasetfromdiskandperformpreprocessing
toprepare
4. itforthelinearclassifier.Thesearethesamestepsaswe
usedforthe
5. SVM,butcondensedtoasinglefunction.
6. """
7.
#LoadtherawCIFAR-10data
8. cifar10_dir='cs231n/datasets/cifar-10-batches-py'
9.
10.
#Cleaningupvariablestopreventloadingdatamultipletime
s(whichmaycausememoryissue)
11. try:
12. delX_train,y_train
13. delX_test,y_test
14. print('Clearpreviouslyloadeddata.')
15. except:
16. pass
17.
18. X_train,y_train,X_test,y_test=load_CIFAR10(cifar10_dir)
19.
20.
#subsamplethedata
21. mask=list(range(num_training,num_training+num_validation)
)
22. X_val=X_train[mask]
23. y_val=y_train[mask]
24. mask=list(range(num_training))
25. X_train=X_train[mask]
26. y_train=y_train[mask]
评论0