kernelmatrix.h

//
// Created by jiashuai on 17-9-19.
//

#ifndef THUNDERSVM_KERNELMATRIX_H
#define THUNDERSVM_KERNELMATRIX_H

#include "thundersvm.h"
#include "syncarray.h"
#include "dataset.h"
#include "svmparam.h"

class KernelMatrix{
public:
    explicit KernelMatrix(const DataSet::node2d &instances, SvmParam param);

    void get_rows(const SyncArray<int> &idx, SyncArray<float_type> &kernel_rows) const;

    void get_rows(const DataSet::node2d &instances, SyncArray<float_type> &kernel_rows) const;

    const SyncArray<float_type> &diag() const;

    size_t n_instances() const { return n_instances_; };

    size_t n_features() const { return n_features_; }

    size_t nnz() const {return nnz_;};//number of nonzero
private:
    KernelMatrix &operator=(const KernelMatrix &) const;

    KernelMatrix(const KernelMatrix &);

    SyncArray<float_type> val_;
    SyncArray<int> col_ind_;
    SyncArray<int> row_ptr_;
    SyncArray<float_type> diag_;
    SyncArray<float_type> self_dot_;
    size_t nnz_;
    size_t n_instances_;
    size_t n_features_;
    SvmParam param;

    void dns_csr_mul(const SyncArray<float_type> &dense_mat, int n_rows, SyncArray<float_type> &result) const;

    void get_dot_product(const SyncArray<int> &idx, SyncArray<float_type> &dot_product) const;

    void get_dot_product(const DataSet::node2d &instances, SyncArray<float_type> &dot_product) const;
};
#endif //THUNDERSVM_KERNELMATRIX_H