File size: 645 Bytes
19b102a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
import numpy as np
class BaseDimensionalityReduction:
""" The Base Dimensionality Reduction class
You can use this to skip over the dimensionality reduction step in BERTopic.
Examples:
This will skip over the reduction step in BERTopic:
```python
from bertopic import BERTopic
from bertopic.dimensionality import BaseDimensionalityReduction
empty_reduction_model = BaseDimensionalityReduction()
topic_model = BERTopic(umap_model=empty_reduction_model)
```
"""
def fit(self, X: np.ndarray = None):
return self
def transform(self, X: np.ndarray) -> np.ndarray:
return X
|