CGAL 5.1.4 - Classification
CGAL::Classification::TensorFlow::Neural_network_classifier< ActivationFunction > Member List

This is the complete list of members for CGAL::Classification::TensorFlow::Neural_network_classifier< ActivationFunction >, including all inherited members.

load_configuration(std::istream &input, bool verbose=false)CGAL::Classification::TensorFlow::Neural_network_classifier< ActivationFunction >
Neural_network_classifier(const Label_set &labels, const Feature_set &features)CGAL::Classification::TensorFlow::Neural_network_classifier< ActivationFunction >
save_configuration(std::ostream &output)CGAL::Classification::TensorFlow::Neural_network_classifier< ActivationFunction >
train(const LabelIndexRange &ground_truth, bool restart_from_scratch=true, std::size_t number_of_iterations=5000, float learning_rate=0.01, std::size_t batch_size=1000, const std::vector< std::size_t > &hidden_layers=std::vector< std::size_t >())CGAL::Classification::TensorFlow::Neural_network_classifier< ActivationFunction >