Dataset Class Reference

Public Member Functions
	Dataset (std::ifstream &infile)
auto	get_input_tensor () const -> Tensor< double, 2 >
auto	get_output_tensor () const -> Tensor< double, 2 >
void	train (NeuralNetwork< double > &net, std::mt19937 &rng) const
auto	test_accuracy (NeuralNetwork< double > &net) const -> double

Constructor & Destructor Documentation

Dataset()

Dataset::Dataset ( std::ifstream & infile )

inlineexplicit

Member Function Documentation

get_input_tensor()

auto Dataset::get_input_tensor ( ) const -> Tensor<double, 2>

inlinenodiscard

Here is the caller graph for this function:

get_output_tensor()

auto Dataset::get_output_tensor ( ) const -> Tensor<double, 2>

inlinenodiscard

Here is the caller graph for this function:

test_accuracy()

auto Dataset::test_accuracy ( NeuralNetwork< double > & net ) const -> double

inline

Here is the caller graph for this function:

train()

void Dataset::train ( NeuralNetwork< double > & net, std::mt19937 & rng ) const

inline

Here is the call graph for this function:

Here is the caller graph for this function:

---

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

• src/trainer_kan/main.cpp