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(),
|