Bert-base-uncased for Android-Ios Question Classification

Code: See Ainize Workspace <br> Android-Ios-Classification DEMO: Ainize Endpoint <br> Demo web Code: Github <br> Android-Ios-Classification API: Ainize API <br> <br>

Overview

Language model: bert-base-cased <br> Language: English <br> Training data: Question classification Android-Ios dataset from Kaggle

Usage

from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline

model_path = "EasthShin/Android_Ios_Classification"

tokenizer = AutoTokenizer.from_pretrained(model_path)
model = AutoModelForSequenceClassification.from_pretrained(model_path)

classifier = pipeline('text-classification', model=model_path, tokenizer=tokenizer)

question = "I bought goodnote in Appstore"

result = dict()
result[0] = classifier(question)[0]