Skip to content

tagging

WeightedTFIDF

WeightedTFIDF(
    zscore_thresh: float = 2.0,
    score_threshold: float = 0.1,
    weight_col: str | None = None,
    level_col: str | None = None,
)

Bases: BaseEstimator, TransformerMixin

Source code in datasci_toolkit/tagging.py
def __init__(
    self,
    zscore_thresh: float = 2.0,
    score_threshold: float = 0.1,
    weight_col: str | None = None,
    level_col: str | None = None,
) -> None:
    self.zscore_thresh = zscore_thresh
    self.score_threshold = score_threshold
    self.weight_col = weight_col
    self.level_col = level_col