Return the test set.
(self, normalization, cuda)
| 144 | |
| 145 | |
| 146 | def get_test_set(self, normalization, cuda): |
| 147 | """ |
| 148 | Return the test set. |
| 149 | """ |
| 150 | if self.learning_type == "transductive": |
| 151 | return self.stub_sampler(normalization, cuda) |
| 152 | else: |
| 153 | if normalization in self.adj_cache: |
| 154 | r_adj = self.adj_cache[normalization] |
| 155 | else: |
| 156 | r_adj = self._preprocess_adj(normalization, self.adj, cuda) |
| 157 | self.adj_cache[normalization] = r_adj |
| 158 | fea = self._preprocess_fea(self.features, cuda) |
| 159 | return r_adj, fea |
| 160 | |
| 161 | def get_val_set(self, normalization, cuda): |
| 162 | """ |
no test coverage detected