beny2000/product_store_type_classification
Preview • Updated • 7 • 1
How to use beny2000/store_type_classifyer with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="beny2000/store_type_classifyer") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("beny2000/store_type_classifyer")
model = AutoModelForSequenceClassification.from_pretrained("beny2000/store_type_classifyer")This model classifies product names by store types from Google's Places API
Accuracy: 69%
LABEL_0: atm
LABEL_1: bakery
LABEL_2: beauty_salon
LABEL_3: bicycle_store
LABEL_4: book_store
LABEL_5: cafe
LABEL_6: car_repair
LABEL_7: car_wash
LABEL_8: clothing_store
LABEL_9: convenience_store
LABEL_10: department_store
LABEL_11: drugstore
LABEL_12: electronics_store
LABEL_13: florist
LABEL_14: furniture_store
LABEL_15: gas_station
LABEL_16: hair_care
LABEL_17: hardware_store
LABEL_18: home_goods_store
LABEL_19: jewelry_store
LABEL_20: liquor_store
LABEL_21: pet_store
LABEL_22: pharmacy
LABEL_23: shoe_store
LABEL_24: supermarket