|
|
#include "utils.hpp" |
|
|
#include <iostream> |
|
|
#include <fstream> |
|
|
#include <stdexcept> |
|
|
#include <iomanip> |
|
|
|
|
|
void check_cuda(cudaError_t st, const char* msg) { |
|
|
if (st != cudaSuccess) { |
|
|
std::cerr << "[CUDA ERROR] " << msg << ": " << cudaGetErrorString(st) << "\n"; |
|
|
std::exit(1); |
|
|
} |
|
|
} |
|
|
|
|
|
void write_submission_csv(const std::string& path, |
|
|
const std::vector<std::string>& ids, |
|
|
const std::vector<float>& probabilities) { |
|
|
std::ofstream ofs(path); |
|
|
if (!ofs.is_open()) { |
|
|
throw std::runtime_error("Cannot open submission file: " + path); |
|
|
} |
|
|
ofs << "id,label\n"; |
|
|
ofs << std::fixed << std::setprecision(8); |
|
|
for (size_t i = 0; i < ids.size(); ++i) { |
|
|
ofs << ids[i] << "," << probabilities[i] << "\n"; |
|
|
} |
|
|
ofs.close(); |
|
|
std::cout << "[INFO] Submission saved to: " << path << " (" << ids.size() << " rows)\n"; |
|
|
}
|
|
|
|
