(self, data, batch=1000, n_neighbors=5)
| 71 | self.fitted = False |
| 72 | |
| 73 | def fit(self, data, batch=1000, n_neighbors=5): |
| 74 | self.data = data |
| 75 | self.embeddings = self.get_text_embedding(data, batch=batch) |
| 76 | n_neighbors = min(n_neighbors, len(self.embeddings)) |
| 77 | self.nn = NearestNeighbors(n_neighbors=n_neighbors) |
| 78 | self.nn.fit(self.embeddings) |
| 79 | self.fitted = True |
| 80 | |
| 81 | def __call__(self, text, return_data=True): |
| 82 | inp_emb = self.use([text]) |
no test coverage detected