prediction.h

Go to the documentation of this file.

// Copyright (c) 2021, Aalto University, developed by Erik Schultheis
// All rights reserved.
//
// SPDX-License-Identifier: MIT
 
#ifndef DISMEC_IO_PREDICTION_H
#define DISMEC_IO_PREDICTION_H
 
#include <filesystem>
#include "matrix_types.h"
 
namespace dismec::io::prediction
{
    using std::filesystem::path;
    void save_sparse_predictions(const path& target,
                                 const PredictionMatrix& values,
                                 const IndexMatrix& indices);
 
    void save_sparse_predictions(std::ostream& target,
                                 const PredictionMatrix& values,
                                 const IndexMatrix& indices);
 
    std::pair<IndexMatrix, PredictionMatrix> read_sparse_prediction(std::istream& source);
 
    std::pair<IndexMatrix, PredictionMatrix> read_sparse_prediction(const path& source);
 
    void save_dense_predictions_as_txt(const path& target, const PredictionMatrix& values);
 
    void save_dense_predictions_as_txt(std::ostream& target, const PredictionMatrix& values);
    void save_dense_predictions_as_npy(const path& target, const PredictionMatrix& values);
 
    void save_dense_predictions_as_npy(std::ostream& target, const PredictionMatrix& values);
}
 
#endif //DISMEC_IO_PREDICTION_H