nfhakim's picture
Update README.md
66d40eb verified
|
raw
history blame
696 Bytes
metadata
language:
  - id
tags:
  - indobert
  - indobenchmark
  - indonlu

This is the second classification of sentiment analysis for (redacted) task

How to import

import torch
from transformers import BertForSequenceClassification, BertTokenizer, BertConfig

tokenizer = BertTokenizer.from_pretrained("nfhakim/sentiment-analysis-c2")
config = BertConfig.from_pretrained("nfhakim/sentiment-analysis-c2")
model = BertForSequenceClassification.from_pretrained("nfhakim/sentiment-analysis-c2", config=config)

How to use

from transformers import pipeline
nlp = pipeline("text-classification", model="nfhakim/sentiment-analysis-c2")

results = nlp("Your input text here")