SushantGautam commited on
Commit
ca59e56
·
1 Parent(s): d66da66

Remove unused import and update predictions serialization in submission_task2.py

Browse files
medvqa/submission_samples/gi-2025/submission_task2.py CHANGED
@@ -1,4 +1,3 @@
1
- import random
2
  from datasets import Dataset
3
  from sklearn.metrics.pairwise import cosine_similarity
4
  from scipy.linalg import sqrtm
@@ -260,7 +259,7 @@ print(
260
  print("🔍 Saving results to 'predictions_2.json'...")
261
 
262
  output_data = {"submission_info": SUBMISSION_INFO, "public_scores": public_scores, "total_time": total_time, "time_per_item": total_time / len(val_dataset),
263
- "memory_used_mb": final_mem, "model_memory_mb": model_mem_used, "gpu_name": gpu_name, "predictions": all_features, "debug": {
264
  "packages": json.loads(subprocess.check_output([sys.executable, "-m", "pip", "list", "--format=json"])),
265
  "system": {
266
  "python": platform.python_version(),
 
 
1
  from datasets import Dataset
2
  from sklearn.metrics.pairwise import cosine_similarity
3
  from scipy.linalg import sqrtm
 
259
  print("🔍 Saving results to 'predictions_2.json'...")
260
 
261
  output_data = {"submission_info": SUBMISSION_INFO, "public_scores": public_scores, "total_time": total_time, "time_per_item": total_time / len(val_dataset),
262
+ "memory_used_mb": final_mem, "model_memory_mb": model_mem_used, "gpu_name": gpu_name, "predictions": json.dumps(all_features.tolist()), "debug": {
263
  "packages": json.loads(subprocess.check_output([sys.executable, "-m", "pip", "list", "--format=json"])),
264
  "system": {
265
  "python": platform.python_version(),