Bert-base-uncased for Android-Ios Question Classification
Code: See Ainize Workspace
Android-Ios-Classification DEMO: Ainize Endpoint
Demo web Code: Github
Android-Ios-Classification API: Ainize API
Overview
Language model: bert-base-cased
Language: English
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]