marqo-ai / GCL

Generalised Contrastive Learning. This is a Repository for Google Shopping Dataset and Benchmarks followed by our novel fine-grained contrastive learning framework.
https://github.com/marqo-ai
Apache License 2.0
46 stars 2 forks source link

Add test to run evaluation script #3

Closed ishaaq closed 4 months ago

ishaaq commented 4 months ago

Simple test just to sanity check that the eval script is runnable with expected arguments. The test downloads a fixed dataset from S3 to run the evaluation.

Refactored eval_gs_v1.py script to give it an entry function that can be tested.