File size: 2,681 Bytes
b881200 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 |
---
language:
- en
library_name: transformers
tags:
- map
- reviews
- public places
---
# Model Card for Model ID
<!-- Provide a quick summary of what the model is/does. -->
This model predicts the type of a place (e.g. restaurant, hotel, park) based on the text of a user review.
E.g.
'I enjoyed the food, it was very delicious' -> 'Restaurants'
'I liked the exhibition, very inspiring' -> 'Museums and Galleries'
# Model Details
## Model Description
The Bert-User-Review-Rating model is trained on a dataset of 1,300,000 reviews of public places and points of interest. It is capable of classifying the type of a place based on a user review into one of the following categories:
0: 'Specialty Food Stores',
1: 'Hotels and Inns',
2: 'Schools and Universities',
3: 'Shopping mall',
4: 'Museums and Galleries',
5: 'Restaurants',
6: 'Parks',
7: 'Shops',
8: 'Cafes and Coffee Shops',
9: 'Cultural Institutions',
10: 'Places of Worship',
11: 'Leisure and Amusement',
12: 'Tourist Attractions',
13: 'Medical Services',
14: 'Social Services',
15: 'Food Courts',
16: 'Sports and Fitness',
17: 'Outdoor Activities',
18: 'Training and Development',
19: 'Bars and Pubs',
20: 'Industrial and Commercial',
21: 'Wellness Services',
22: 'Pets Services',
23: 'Public Transit',
24: 'Performing Arts',
25: 'Vehicle Services',
26: 'Other Lodging',
27: 'Professional Services',
28: 'Government Services',
29: 'Religious Services',
30: 'Travel Services'
Model type: BERT-based model
Language(s) (NLP): English
# Direct Use
<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
The model can be used directly to classify the type of a place based on a user review.
# Bias, Risks and Limitations
The model may reflect biases present in the training data, such as cultural or regional biases, as training data reflects public places in Singapore.
# How to Get Started with the Model
Use the code below to get started with the model.
## Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="mekes/Bert-Place-Type")
result = pipe("The food was super tasty, I enjoyed every bite.")
print(result)
# Metrics
<!-- These are the evaluation metrics being used, ideally with a description of why. -->
Eval Accuracy: 0.753
Eval F1 Score: 0.741
Eval Recall: 0.753
# Environmental Impact
Carbon emissions were estimated using the Machine Learning Impact calculator presented in Lacoste et al. (2019)
Calculations were done with Nvidia RTX 3090 instead of the used Nvidia RTX 4090.
For one training run it emmited approximately 1,5 kg CO2
|