# [ENAP] BERT para classificação — Data

来源：https://www.kaggle.com/competitions/enap-bert-para-classificao/data

本地 Meta Kaggle 快照；不是本次实时页面。

# Dataset Description

## Base de treinamento

A base de treinamento contém o `label` e o `text` em uma lista. 

| label | text                                                  |
|-------|-------------------------------------------------------|
| 4     | dr. goldberg offers everything i look for in a...     |
| 1     | Unfortunately, the frustration of being Dr. Go...     |
| 3     | Been going to Dr. Goldberg for over 10 years. ...     |
| 3     | Got a letter in the mail last week that said D...     |
| 0     | I don't know what Dr. Goldberg was like before...     |

<br>A base de treinamento pode ser acessada com as seguintes linhas de código:

```python
!pip install -q datasets 
from datasets import load_dataset
train = load_dataset('yelp_review_full', split='train')
```

## Base de teste

A base de teste contém o `ID` e o `text` em uma lista. 

| ID    | text                                               |
|-------|----------------------------------------------------|
| 30978 | Stopped in for a drink before a show at the Ca...  |
| 18993 | Why am I reviewing this place? Because upon a...   |
| 1311  | I came to Souper Salad right after getting rej...  |
| 49832 | We went to this Bistro based on a recommendati...  |
| 43775 | Took my first class today, the class was aweso...  |

<br>A base de teste pode ser lida com as seguintes linhas de código:

```python
import pandas as pd
from datasets import Dataset

test = pd.read_csv('test.csv')
test = Dataset.from_pandas(test)
```