(self, length, save_path=None)
| 210 | class ClassificationPredSaver(object): |
| 211 | |
| 212 | def __init__(self, length, save_path=None): |
| 213 | |
| 214 | if save_path is not None: |
| 215 | |
| 216 | # Remove filetype from save_path |
| 217 | save_path = save_path.split('.')[0] |
| 218 | self.save_path = save_path |
| 219 | |
| 220 | self.length = length |
| 221 | |
| 222 | self.all_preds = None |
| 223 | self.all_labels = None |
| 224 | |
| 225 | self.running_start_idx = 0 |
| 226 | |
| 227 | def update(self, preds, labels=None): |
| 228 |
nothing calls this directly
no outgoing calls
no test coverage detected