Cosan::RandomKFoldParallel Class Reference

#include <randomkfold.h>

Inheritance diagram for Cosan::RandomKFoldParallel:

Public Member Functions
	RandomKFoldParallel ()
	RandomKFoldParallel (gsl::index kfoldnumber)
	RandomKFoldParallel (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 43 of file randomkfold.h.

Constructor & Destructor Documentation

RandomKFoldParallel() [1/3]

Cosan::RandomKFoldParallel::RandomKFoldParallel ( )

inline

Definition at line 45 of file randomkfold.h.

: Splitter() {}

RandomKFoldParallel() [2/3]

Cosan::RandomKFoldParallel::RandomKFoldParallel ( gsl::index  kfoldnumber )

inline

Definition at line 46 of file randomkfold.h.

: Splitter(kfoldnumber) {}

RandomKFoldParallel() [3/3]

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

inline

Definition at line 47 of file randomkfold.h.

: Splitter(nrows,kfoldnumber){}

Member Function Documentation

GetSplit()

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

inline

Definition at line 73 of file randomkfold.h.

{return split_batch;}

SetSplit()

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

inlinevirtual

Reimplemented from Cosan::Splitter.

Definition at line 48 of file randomkfold.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,trainidx;
                std::sample(idx.begin(), idx.end(), std::back_inserter(testidx),
                            foldSize, std::mt19937{std::random_device{}()});
                std::sort(testidx.begin(),testidx.end());
                std::set_difference(idx.begin(), idx.end(), testidx.begin(), testidx.end(),
                                    std::inserter(trainidx, trainidx.begin()));
                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(),trainidx.size(),testidx.size());
//                mylock.lock();
                split_batch[i] = {trainidx,testidx};
//                mylock.unlock();
            }
        }

Member Data Documentation

split_batch

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

private

Definition at line 75 of file randomkfold.h.

---

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

• cosan/selection/randomkfold.h