LiYuan
/

amazon-query-product-ranking

Text Classification Transformers PyTorch TensorBoard distilbert generated_from_trainer Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

LiYuan commited on Apr 28, 2022

Commit

3496770

•

1 Parent(s): fdb3b5f

Update

Files changed (1) hide show

README.md +4 -2

README.md CHANGED Viewed

@@ -51,8 +51,10 @@ model = AutoModelForSequenceClassification.from_pretrained("LiYuan/amazon-query-
 Download all the raw [dataset](https://www.aicrowd.com/challenges/esci-challenge-for-improving-product-search/dataset_files) from the Amazon KDD Cup website.
-The three dataset we used like below:
 ## Training procedure

 Download all the raw [dataset](https://www.aicrowd.com/challenges/esci-challenge-for-improving-product-search/dataset_files) from the Amazon KDD Cup website.
+1. Concatenate the all product attributes from the product dataset
+2. Join it with a training query dataset
+3. Stratified Split the merged data into 571,223-row training, 142,806-row validation, 79,337-row test set
+4. Train on the full training set
 ## Training procedure