DiSMEC++
training.h File Reference
#include "spec.h"
#include "parallel/task.h"
#include "solver/minimizer.h"
#include <memory>
#include <functional>
#include "fwd.h"
#include "data/types.h"

Go to the source code of this file.

Classes

class  dismec::TrainingTaskGenerator
 Generates tasks for training weights for the i'th label. More...
 
struct  dismec::TrainingResult
 

Namespaces

 dismec
 Main namespace in which all types, classes, and functions are defined.
 

Functions

TrainingResult dismec::run_training (parallel::ParallelRunner &runner, std::shared_ptr< TrainingSpec > spec, label_id_t begin_label=label_id_t{0}, label_id_t end_label=label_id_t{-1})