├── LICENSE ├── MANIFEST.in ├── README.md ├── datasets ├── BC5CDR │ ├── dev.txt │ ├── labels.txt │ └── train.txt ├── BIOSSES │ ├── dev.tsv │ └── train.tsv ├── DiabetesNER │ ├── dev.txt │ ├── labels.txt │ └── train.txt └── TrialClassification │ ├── dev.tsv │ └── train.tsv ├── requirements.txt ├── setup.py ├── transformers_sklearn ├── __init__.py ├── classification.py ├── classification_sda.py ├── extract_features.py ├── freelb_classification.py ├── model_albert │ ├── __init__.py │ ├── configuration_bert.py │ ├── configuration_utils.py │ ├── file_utils.py │ ├── modeling_albert.py │ ├── modeling_utils.py │ ├── optimization.py │ ├── tokenization_bert.py │ └── tokenization_utils.py ├── model_albert_fix │ ├── __init__.py │ ├── configuration_albert.py │ ├── configuration_bert.py │ ├── configuration_utils.py │ ├── file_utils.py │ ├── modeling_albert.py │ ├── modeling_albert_bright.py │ ├── modeling_bert.py │ ├── modeling_utils.py │ ├── tokenization_albert.py │ ├── tokenization_bert.py │ └── tokenization_utils.py ├── model_el │ ├── __init__.py │ ├── el_utils.py │ ├── model.py │ └── use_api.py ├── model_electra │ ├── __init__.py │ ├── configuration_electra.py │ ├── configuration_utils.py │ ├── convert_electra_original_tf_checkpoint_to_pytorch.py │ ├── file_utils.py │ ├── modeling_electra.py │ └── tokenization_electra.py ├── model_summarization │ ├── __init__.py │ ├── file_utils.py │ ├── modeling_summarization.py │ └── modeling_utils.py ├── regression.py ├── summarization.py ├── token_classification.py └── utils │ ├── __init__.py │ ├── classification_utils.py │ ├── data_utils.py │ ├── features_utils.py │ ├── loss_utils.py │ ├── metrics_utils.py │ ├── regression_utils.py │ ├── summarization_utils.py │ └── token_classification_utils.py └── tutorials ├── bc5cdr.py ├── biosses.py ├── diabetes_ner.py ├── gmbner.py ├── lcqmc.py ├── mrpc.py ├── msraner.py ├── sts-b.py └── trail.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- 1 | include LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/README.md -------------------------------------------------------------------------------- /datasets/BC5CDR/dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/datasets/BC5CDR/dev.txt -------------------------------------------------------------------------------- /datasets/BC5CDR/labels.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/datasets/BC5CDR/labels.txt -------------------------------------------------------------------------------- /datasets/BC5CDR/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/datasets/BC5CDR/train.txt -------------------------------------------------------------------------------- /datasets/BIOSSES/dev.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/datasets/BIOSSES/dev.tsv -------------------------------------------------------------------------------- /datasets/BIOSSES/train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/datasets/BIOSSES/train.tsv -------------------------------------------------------------------------------- /datasets/DiabetesNER/dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/datasets/DiabetesNER/dev.txt -------------------------------------------------------------------------------- /datasets/DiabetesNER/labels.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/datasets/DiabetesNER/labels.txt -------------------------------------------------------------------------------- /datasets/DiabetesNER/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/datasets/DiabetesNER/train.txt -------------------------------------------------------------------------------- /datasets/TrialClassification/dev.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/datasets/TrialClassification/dev.tsv -------------------------------------------------------------------------------- /datasets/TrialClassification/train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/datasets/TrialClassification/train.tsv -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/setup.py -------------------------------------------------------------------------------- /transformers_sklearn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/__init__.py -------------------------------------------------------------------------------- /transformers_sklearn/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/classification.py -------------------------------------------------------------------------------- /transformers_sklearn/classification_sda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/classification_sda.py -------------------------------------------------------------------------------- /transformers_sklearn/extract_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/extract_features.py -------------------------------------------------------------------------------- /transformers_sklearn/freelb_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/freelb_classification.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert/__init__.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert/configuration_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert/configuration_bert.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert/configuration_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert/configuration_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert/file_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert/modeling_albert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert/modeling_albert.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert/modeling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert/modeling_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert/optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert/optimization.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert/tokenization_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert/tokenization_bert.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert/tokenization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert/tokenization_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert_fix/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert_fix/__init__.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert_fix/configuration_albert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert_fix/configuration_albert.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert_fix/configuration_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert_fix/configuration_bert.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert_fix/configuration_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert_fix/configuration_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert_fix/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert_fix/file_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert_fix/modeling_albert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert_fix/modeling_albert.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert_fix/modeling_albert_bright.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert_fix/modeling_albert_bright.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert_fix/modeling_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert_fix/modeling_bert.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert_fix/modeling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert_fix/modeling_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert_fix/tokenization_albert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert_fix/tokenization_albert.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert_fix/tokenization_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert_fix/tokenization_bert.py -------------------------------------------------------------------------------- /transformers_sklearn/model_albert_fix/tokenization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_albert_fix/tokenization_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/model_el/__init__.py: -------------------------------------------------------------------------------- 1 | from .use_api import BERTologyELClassifier -------------------------------------------------------------------------------- /transformers_sklearn/model_el/el_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_el/el_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/model_el/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_el/model.py -------------------------------------------------------------------------------- /transformers_sklearn/model_el/use_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_el/use_api.py -------------------------------------------------------------------------------- /transformers_sklearn/model_electra/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_electra/__init__.py -------------------------------------------------------------------------------- /transformers_sklearn/model_electra/configuration_electra.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_electra/configuration_electra.py -------------------------------------------------------------------------------- /transformers_sklearn/model_electra/configuration_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_electra/configuration_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/model_electra/convert_electra_original_tf_checkpoint_to_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_electra/convert_electra_original_tf_checkpoint_to_pytorch.py -------------------------------------------------------------------------------- /transformers_sklearn/model_electra/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_electra/file_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/model_electra/modeling_electra.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_electra/modeling_electra.py -------------------------------------------------------------------------------- /transformers_sklearn/model_electra/tokenization_electra.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_electra/tokenization_electra.py -------------------------------------------------------------------------------- /transformers_sklearn/model_summarization/__init__.py: -------------------------------------------------------------------------------- 1 | from .modeling_summarization import BeamSearch -------------------------------------------------------------------------------- /transformers_sklearn/model_summarization/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_summarization/file_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/model_summarization/modeling_summarization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_summarization/modeling_summarization.py -------------------------------------------------------------------------------- /transformers_sklearn/model_summarization/modeling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/model_summarization/modeling_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/regression.py -------------------------------------------------------------------------------- /transformers_sklearn/summarization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/summarization.py -------------------------------------------------------------------------------- /transformers_sklearn/token_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/token_classification.py -------------------------------------------------------------------------------- /transformers_sklearn/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/utils/__init__.py -------------------------------------------------------------------------------- /transformers_sklearn/utils/classification_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/utils/classification_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/utils/data_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/utils/features_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/utils/features_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/utils/loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/utils/loss_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/utils/metrics_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/utils/metrics_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/utils/regression_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/utils/regression_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/utils/summarization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/utils/summarization_utils.py -------------------------------------------------------------------------------- /transformers_sklearn/utils/token_classification_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/transformers_sklearn/utils/token_classification_utils.py -------------------------------------------------------------------------------- /tutorials/bc5cdr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/tutorials/bc5cdr.py -------------------------------------------------------------------------------- /tutorials/biosses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/tutorials/biosses.py -------------------------------------------------------------------------------- /tutorials/diabetes_ner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/tutorials/diabetes_ner.py -------------------------------------------------------------------------------- /tutorials/gmbner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/tutorials/gmbner.py -------------------------------------------------------------------------------- /tutorials/lcqmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/tutorials/lcqmc.py -------------------------------------------------------------------------------- /tutorials/mrpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/tutorials/mrpc.py -------------------------------------------------------------------------------- /tutorials/msraner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/tutorials/msraner.py -------------------------------------------------------------------------------- /tutorials/sts-b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/tutorials/sts-b.py -------------------------------------------------------------------------------- /tutorials/trail.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/trueto/transformers_sklearn/HEAD/tutorials/trail.py --------------------------------------------------------------------------------