Cosan::KFold Class Reference

#include <kfold.h>

Inheritance diagram for Cosan::KFold:

Public Member Functions
	KFold ()
	KFold (gsl::index kfoldnumber)
	KFold (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

https://en.wikipedia.org/wiki/Cross-validation_(statistics)

Definition at line 12 of file kfold.h.

Constructor & Destructor Documentation

KFold() [1/3]

Cosan::KFold::KFold ( )

inline

Definition at line 14 of file kfold.h.

: Splitter() {}

KFold() [2/3]

Cosan::KFold::KFold ( gsl::index  kfoldnumber )

inline

Definition at line 15 of file kfold.h.

: Splitter(kfoldnumber) {}

KFold() [3/3]

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

inline

Definition at line 16 of file kfold.h.

: Splitter(nrows,kfoldnumber){}

Member Function Documentation

GetSplit()

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

inline

Definition at line 65 of file kfold.h.

{ return split_batch; }

SetSplit()

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

inlinevirtual

Reimplemented from Cosan::Splitter.

Definition at line 17 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;
            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 {:}, trainidx size:{:}, testidx size:{:}\n",
                           i, 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.push_back({trainidx, testidx});
 
            }
        }

Member Data Documentation

split_batch

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

private

Definition at line 69 of file kfold.h.

---

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

• cosan/selection/kfold.h