| 157 | return self |
| 158 | |
| 159 | def jira_issue_transform(self, project_list, use_description, use_comment): |
| 160 | num_samples = sum([ |
| 161 | sum([1 for issue in self.datasets[fs] if self.file_filter(issue)]) |
| 162 | for fs in project_list]) |
| 163 | |
| 164 | features = {} |
| 165 | features['text'] = [None] * num_samples |
| 166 | labels = [None] * num_samples |
| 167 | ind = 0 |
| 168 | for project in project_list: |
| 169 | for issue_id, issue in self.datasets[project].items(): |
| 170 | if self.file_filter(issue): |
| 171 | features['text'][ind] = issue[self.text_feature] |
| 172 | if use_description and 'description' not in self.text_feature: |
| 173 | features['text'][ind] = issue[self.text_feature] + issue['description'].strip() |
| 174 | if use_comment and 'comment' not in self.text_feature: |
| 175 | features['text'][ind] = issue[self.text_feature] + issue['comment'].strip() |
| 176 | if use_description and use_comment: |
| 177 | features['text'][ind] = issue[self.text_feature] + issue['description'].strip() + issue['comment'].strip() |
| 178 | labels[ind] = self.label_func(issue, self.is_jira) |
| 179 | ind += 1 |
| 180 | return features, labels |
| 181 | |
| 182 | def fs_patch_transform(self, fs_list): |
| 183 | num_samples = sum([ |