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]