with OUTPUT_PATH.open("w", newline="") as handle:
    writer = csv.writer(handle)
    writer.writerow([ID_COLUMN, "score"])
    for account_id, score in zip(account_ids, prediction_scores):
        writer.writerow([account_id, f"{score:.6f}"])

print(f"Model: {MODEL_PATH}")
print(f"Input rows: {len(input_rows)}")
print(f"Prediction shape: {prediction_array.shape}")
print(f"Output: {OUTPUT_PATH}")