bertstem 0.0.9

BERT model fine-tuned on chilean STEM lessons

BERT-STEM

BERT model fine-tuned on Science Technology Engineering and Mathematics (STEM) lessons.

Install:

To install from pip:

pip install bertstem

Quickstart

To use it:

from BERT_STEM.Encode import *
import pandas as pd
import transformers

# Download spanish BERT:
model = transformers.BertModel.from_pretrained("dccuchile/bert-base-spanish-wwm-uncased")

# Download spanish tokenizer:
tokenizer = transformers.BertTokenizerFast.from_pretrained("dccuchile/bert-base-spanish-wwm-uncased",
                                                            do_lower_case=True, 
                                                            add_special_tokens = False)

# Example dataframe with text in spanish
data = {'col_1': [3, 2, 1], 
        'col_2': ['hola como estan', 'alumnos queridos', 'vamos a hablar de matematicas']}
        
df = pd.DataFrame.from_dict(data)

# Encode sentences using spanish BERT:
sentence_encoder(df, model, tokenizer, column = 'col_2', encoding = 'sum')