Searched refs:example_label (Results 1 - 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | loss.h | 37 const double example_label, 42 virtual double ComputePrimalLoss(const double wx, const double example_label, 47 const double example_label, 55 virtual Status ConvertLabel(float* const example_label) const = 0;
|
H A D | hinge-loss.h | 75 double ComputeDualLoss(const double current_dual, const double example_label, 79 const double y_alpha = current_dual * example_label; // y \alpha 89 double ComputePrimalLoss(const double wx, const double example_label, 91 const double y_wx = example_label * wx; 109 Status ConvertLabel(float* const example_label) const final { 110 if (*example_label == 0.0) { 111 *example_label = -1; 114 if (*example_label == 1.0) { 120 *example_label);
|
H A D | smooth-hinge-loss.h | 56 double ComputeDualLoss(const double current_dual, const double example_label, 60 const double y_alpha = current_dual * example_label; // y \alpha 68 double ComputePrimalLoss(const double wx, const double example_label, 70 const double y_wx = example_label * wx; 78 Status ConvertLabel(float* const example_label) const final { 79 if (*example_label == 0.0) { 80 *example_label = -1; 83 if (*example_label == 1.0) { 89 *example_label);
|
H A D | squared-loss.h | 44 double ComputeDualLoss(const double current_dual, const double example_label, 48 return current_dual * (0.5 * current_dual - example_label) * example_weight; 52 double ComputePrimalLoss(const double wx, const double example_label, 54 const double error = wx - example_label; 66 Status ConvertLabel(float* const example_label) const final {
|
H A D | logistic-loss.h | 48 double ComputeDualLoss(const double current_dual, const double example_label, 52 const double ay = current_dual * example_label; 61 double ComputePrimalLoss(const double wx, const double example_label, 68 const double y_wx = example_label * wx; 99 Status ConvertLabel(float* const example_label) const final { 100 if (*example_label == 0.0) { 101 *example_label = -1; 104 if (*example_label == 1.0) { 110 *example_label);
|
H A D | loss_test.cc | 190 float example_label = 1.0; local 194 TF_EXPECT_OK(loss_updater.ConvertLabel(&example_label)); 195 EXPECT_EQ(1.0, example_label); 198 example_label = 0.0; 199 TF_EXPECT_OK(loss_updater.ConvertLabel(&example_label)); 200 EXPECT_EQ(-1.0, example_label); 203 example_label = 0.5; 204 status = loss_updater.ConvertLabel(&example_label);
|
H A D | sdca_ops.cc | 170 float example_label = example.example_label(); 172 options.loss_updater->ConvertLabel(&example_label); 190 options.num_loss_partitions, example_label, example_weight, dual, 205 example_statistics.prev_wx[0], example_label, example_weight); 207 options.loss_updater->ComputeDualLoss(dual, example_label,
|
H A D | sdca_internal.h | 140 float example_label() const { return example_label_; } function in class:tensorflow::sdca::Example
|
H A D | sdca_internal.cc | 244 float label = example.example_label();
|
Completed in 509 milliseconds