EasthShin/Android_Ios_Classification

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]