...
Code Block | ||||
---|---|---|---|---|
| ||||
from .latent_semantic_indexing import LatentSemanticIndexing from .bert import Bert from .bert_qa import BertQA from .sbert import SBert from .sentiment_analysis_vader import SentimentAnalysisVader from .sentiment_analysis_text_blob import SentimentAnalysisTextBlob from .model_wrapper import ModelWrapper from .tfidf_vectorizer import TfidfVectorizer from .test import Test |
config/config.json
...
Code Block | ||||
---|---|---|---|---|
| ||||
"model_types": { "LatentSemanticIndexing" : { "enabled": true, "input_data_as_tokens": false, "model_names": ["lsi"] }, "Bert": { "enabled": false, "input_data_as_tokens": false, "model_names": ["bert-base-uncased"], "default_model": "bert-base-uncased" }, "BertQA": { "enabled": false, "input_data_as_tokens": false, "model_names": ["bert-large-uncased-whole-word-masking-finetuned-squad"], "default_model": "bert-large-uncased-whole-word-masking-finetuned-squad" }, "SBert": { "enabled": true, "input_data_as_tokens": false, "model_names": ["bert-base-nli-stsb-mean-tokens", "bert-base-nli-mean-tokens", "distilbert-base-nli-stsb-mean-tokens"], "default_model": "bert-base-nli-stsb-mean-tokens" }, "SentimentAnalysisVader": { "enabled": true, "input_data_as_tokens": false, "model_names": ["vader"] }, "SentimentAnalysisTextBlob": { "enabled": true, "input_data_as_tokens": false, "model_names": ["textBlob"] }, "TfidfVectorizer": { "enabled": true, "input_data_as_tokens": false, "model_names": ["Tfidf"] }, "Test" { "enabled": true, "input_data_as_tokens": false, "model_names": ["test"] } } |
...