Cosan::KFoldParallel Class Reference

#include <kfold.h>

Inheritance diagram for Cosan::KFoldParallel:

Public Member Functions
	KFoldParallel ()
	KFoldParallel (gsl::index kfoldnumber)
	KFoldParallel (gsl::index nrows, gsl::index kfoldnumber)
void	SetSplit (gsl::index nrows)
std::vector< std::tuple< std::vector< gsl::index >, std::vector< gsl::index > > > &	GetSplit ()
Public Member Functions inherited from Cosan::Splitter
	Splitter ()
	Splitter (gsl::index kfoldnumber)
	Splitter (gsl::index nrows, gsl::index kfoldnumber)
gsl::index	GetKFoldNumber ()
Public Member Functions inherited from Cosan::Selection
	Selection ()
Public Member Functions inherited from Cosan::CosanBO
	CosanBO ()
	Default constructor. More...
virtual const std::string	GetName () const
	Get the name of the objects. More...

Private Attributes
std::vector< std::tuple< std::vector< gsl::index >, std::vector< gsl::index > > >	split_batch

Additional Inherited Members
Protected Attributes inherited from Cosan::Splitter
gsl::index	KFoldNumber = 5

Detailed Description

Definition at line 143 of file kfold.h.

Constructor & Destructor Documentation

KFoldParallel() [1/3]

Cosan::KFoldParallel::KFoldParallel ( )

inline

Definition at line 145 of file kfold.h.

: Splitter() {}

KFoldParallel() [2/3]

Cosan::KFoldParallel::KFoldParallel ( gsl::index  kfoldnumber )

inline

Definition at line 146 of file kfold.h.

: Splitter(kfoldnumber) {}

KFoldParallel() [3/3]

Cosan::KFoldParallel::KFoldParallel ( gsl::index  nrows, gsl::index  kfoldnumber  )

inline

Definition at line 147 of file kfold.h.

: Splitter(nrows,kfoldnumber){}

Member Function Documentation

GetSplit()

std::vector<std::tuple<std::vector < gsl::index>, std::vector<gsl::index> > >& Cosan::KFoldParallel::GetSplit ( )

inline

Definition at line 172 of file kfold.h.

{ return split_batch; }

SetSplit()

void Cosan::KFoldParallel::SetSplit ( gsl::index  nrows )

inlinevirtual

Reimplemented from Cosan::Splitter.

Definition at line 148 of file kfold.h.

                                     {
            if (nrows <= KFoldNumber) {
                throw SmallRows;
            }
            std::vector <gsl::index> idx(nrows);
            std::iota(idx.begin(), idx.end(), 0);
            gsl::index foldSize = nrows / KFoldNumber;
//            std::mutex mylock;
            split_batch.resize(KFoldNumber);
            #pragma omp parallel for
            for (gsl::index i = 0; i < KFoldNumber; i++) {
                std::vector <gsl::index> testidx(foldSize), trainidx;
                if (i == KFoldNumber - 1) {
                    testidx.resize(nrows - i * foldSize);
                }
                fmt::print(
                        "Current Index is {:}, the current thread num is {:}, total number of threads {:}. trainidx size:{:}, testidx size:{:}\n",
                        i, omp_get_thread_num(), omp_get_num_threads(), nrows - foldSize, foldSize);
                std::iota(testidx.begin(), testidx.end(), i * foldSize);
                std::set_difference(idx.begin(), idx.end(), testidx.begin(), testidx.end(),
                                    std::inserter(trainidx, trainidx.begin()));
                split_batch[i] = {trainidx, testidx};
            }
        }

Member Data Documentation

split_batch

std::vector<std::tuple<std::vector < gsl::index>, std::vector<gsl::index> > > Cosan::KFoldParallel::split_batch

private

Definition at line 175 of file kfold.h.

---

The documentation for this class was generated from the following file:

• cosan/selection/kfold.h