1/Update setup.py to support following modes:

- default (most frequent mode)
- torch
- transformers
- all
2/ Update Docker file
3/ Update documentation as well.
This commit is contained in:
unclecode
2024-06-30 00:15:29 +08:00
parent 5b28eed2c0
commit 61ae2de841
3 changed files with 59 additions and 37 deletions

View File

@@ -10,7 +10,7 @@ from functools import partial
from .model_loader import *
import math
import numpy as np
class ExtractionStrategy(ABC):
"""
Abstract base class for all extraction strategies.
@@ -219,6 +219,8 @@ class CosineStrategy(ExtractionStrategy):
"""
super().__init__()
import numpy as np
self.semantic_filter = semantic_filter
self.word_count_threshold = word_count_threshold
self.max_dist = max_dist