| 258 | return network_weights |
| 259 | |
| 260 | class MyDataset(TensorDataset): |
| 261 | def __init__(self, trainval_df): |
| 262 | super(MyDataset, self).__init__() |
| 263 | self.df = trainval_df |
| 264 | self.df.reset_index(drop=True, inplace=True) # train_test_split之后，数据集的index混乱，需要reset |
| 265 | |
| 266 | def __len__(self): |
| 267 | return len(self.df) |
| 268 | |
| 269 | def __getitem__(self, index): |
| 270 | |
| 271 | #drug1, drug2, cell, target |
| 272 | return (self.df.loc[index,0], self.df.loc[index,1], self.df.loc[index,2], self.df.loc[index,3], self.df.loc[index,4]) |
| 273 | |
| 274 | def k_fold_trainer_graph(temp_loader_trainval,model,args): |
| 275 | |