Code to submit predictions to cardinality estimation leaderboard @ ML for Systems NeurIPS 2021 workshop

save_submission.py

import numpy as np
def save_submission(predictions, filename):
    "Take model output & save for cardinality estimation benchmark upload."""
    np.save(filename, np.array([value for item in predictions for key, value in sorted(item.items())]))
# preds = rf.test(testqs) # run your prediction code on the test data
save_submission(preds, "mysubmission.npy")
# Then Upload "mysubmission.npy" to the leaderboard https://mlforsystems.wl.r.appspot.com :)