SciTensor commited on
Commit
f85306e
·
verified ·
1 Parent(s): 1a32978

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +68 -3
README.md CHANGED
@@ -1,3 +1,68 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - uk
5
+ - en
6
+ base_model:
7
+ - intfloat/multilingual-e5-base
8
+ ---
9
+ # Model Card for Retail Product Title Classifier (E5 fine-tuned)
10
+
11
+ ## Model Details
12
+
13
+ ### Model Description
14
+
15
+ A fine-tuned version of `intfloat/multilingual-e5-base`, adapted for the classification of retail product titles in Ukrainian and English.
16
+ The model is optimized for noisy, real-world data (e.g., typos, abbreviations) typically encountered in e-commerce catalogues.
17
+
18
+ - **Developed by:** Viacheslav Trachov
19
+ - **Model type:** Transformer Encoder (E5)
20
+ - **Language(s):** Ukrainian, English
21
+ - **License:** Apache 2.0
22
+ - **Finetuned from model:** intfloat/multilingual-e5-base
23
+
24
+ ## Uses
25
+
26
+ ### Direct Use
27
+
28
+ - Classifying short, noisy product titles into predefined retail categories.
29
+ - Designed for retail inventory management, e-commerce catalogues, and internal search optimization.
30
+
31
+ ### Out-of-Scope Use
32
+
33
+ - Free-text generation or long-form document classification.
34
+ - Tasks requiring high performance on languages other than Ukrainian/English.
35
+
36
+ ## Bias, Risks, and Limitations
37
+
38
+ - Performance may degrade on titles that mix multiple languages or are heavily abbreviated beyond retail-specific contexts.
39
+ - Categories must match the domain and fine-tuning setup (i.e., Ukrainian e-commerce retail).
40
+
41
+ ### Recommendations
42
+
43
+ - Use confidence thresholds to route low-confidence predictions for manual review if critical.
44
+ - Test on domain-specific datasets if adapting to new industries.
45
+
46
+ ## Training Details
47
+
48
+ ### Training Data
49
+ - ~60,000 real-world Ukrainian product titles from an e-commerce aggregator.
50
+ - Titles were preprocessed minimally (lowercasing, space normalization).
51
+ - Additional synthetic examples were generated for underrepresented categories using ChatGPT-4.
52
+
53
+ ### Training Procedure
54
+ - Finetuned for multi-class classification using Cross-Entropy Loss.
55
+ - Max sequence length: 48 tokens
56
+ - Learning rate: 5e-5
57
+ - Batch size: 64
58
+ - Epochs: 15
59
+
60
+ ### Hardware: NVIDIA V100 GPU
61
+
62
+ ## Evaluation
63
+
64
+ Macro F1-score used due to class imbalance.
65
+ Results:
66
+ macro-F1 (for clean data) 0.830
67
+ macro-F1 (for noisy data) 0.777
68
+ Model achieved strong robustness under simulated typographical noise (~6.3% macro-F1 degradation)