Yelp-Rating-Prediction

作者 SakuraKnight · 下载 1 · 点赞 0 · 更新 2023-11-02

A Demo BERT classification model Trained on (Part of) Yelp Dataset

Photo2Text model: ydshieh/vit-gpt2-coco-en

Expected / Standard Input:

[CLS] Business Name [SEP] Address [SEP] City [SEP] Photo2Text Outputs ...

Example:

[CLS] Paws The Cat Cafe [SEP] 10588 109 Street [SEP] Edmonton [SEP] A cup of coffee

Expected Output: 5