nachiketashunya
Amazon-ML-Challenge-2024
Python

This repo is for Amazon ML Challenge 2024. The challenge was to develop a Machine Learning model to extract product details directly from the product images.

Last updated Aug 8, 2026
86
Stars
25
Forks
0
Issues
+1
Stars/day
Attention Score
29
Language breakdown
Python 99.3%
Dockerfile 0.5%
Jupyter Notebook 0.1%
Shell 0.1%
Makefile 0.0%
โ–ธ Files click to expand
README

Collaborators [Team NeuralNinjas]

Please change the path directories accordingly in each file.

Initial FineTuning the model

  • Download data using the code provided by the host.
  • Preprocess train data preprocess_train.py
  • Creating JSON for finetuning with LLaMA-Factory dataprep.py
  • Follow guidelines given in docs of LLaMA-Factory to register our dataset. (Refer data/example.json)
  • For finetuning follow finetune.ipynb. Settings for finetuning will be registered using WebUI.

Inferencing of Finetuned Model

  • Run inference.py
  • Post Processing postprocess.py
  • Evaluation Metric: F1 Score

Experimentation

Model: https://huggingface.co/Qwen/Qwen2-VL-7B-Instruct
1. Inference Baseline Qwen2VL-7B-Instruct AWQ
        score: 0.617
        prompt: <Return only value>What is the {entity_name} of this product?

2. Finetuning prompt: What is the {entity_name}? a. 10k Samples postprocessing: a.Replaced Range with NA: - Invalid units replaced with NA - score: 0.678 b.Replaced range with Max Value: - score: 0.677

b. 20k Samples with Cosine Scheduler Inference Results: - score: 0.679 FineTuned on Curated 1600 samples: - score: 0.865 - lr_scheduler: reduce-lr-on-plateau FineTuned on 20k samples: - Preprocessing: - Replace Range with Max value - Remove entity values with invalid units

- FineTuned on Curated 1600 Samples: - score: - lr_scheduler: reduce-lr-on-plateau 3. Data Curation - Missing {entity_value} replaced with NA - Correct inaccurate {entity_value} 4. Experimental Settings -- batch_size: 8 -- learning_rate: 5e-5 -- gradient_accumulation: 8 -- scheduler: appropriately choosen -- tool: LLaMA-Factory -- finetuning method: qlora-8bit

laderboard_toppers

ยฉ 2026 GitRepoTrend ยท nachiketashunya/Amazon-ML-Challenge-2024 ยท Updated daily from GitHub