returnout
classtrainingdatasetdataset:
def__init__self:
supertrainingdataset,self.__init__
self.data_dict_xx_train
self.data_dict_yy_train
def__getitem__self,index:
tself.data_dict_x[index,0:36]
,6
returnt,self.data_dict_y[index]
def__len__self:
returnlenself.data_dict_y
classtestdatasetdataset:
def__init__self:
supertestdataset,self.__init__
self.data_dict_xx_validate
self.data_dict_yy_validate
def__getitem__self,index:
tself.data_dict_x[index,0:36]
,6
returnt,self.data_dict_y[index]
def__len__self:
returnlenself.data_dict_y
defcnn_classification:
batch_size256
traindataloaderdataloadertrainingdataset,batch_sizebatch_size,shufflefalse
testdataloaderdataloadertestdataset,batch_sizebatch_size,shufflefalse
epoch_num200
#lr0.001
lr0.001
netvggbasesimple