Program Listing for File train_accuracy.h

↰ Return to documentation for file (include/train_accuracy.h)

#ifndef MINDSPORE_INCLUDE_API_CALLBACK_TRAIN_ACCURACY_H
#define MINDSPORE_INCLUDE_API_CALLBACK_TRAIN_ACCURACY_H

#include <cstddef>
#include <string>
#include <vector>
#include <memory>
#include <utility>
#include "include/api/callback/callback.h"
#include "include/api/metrics/accuracy.h"

using GraphPoint = std::pair<int, float>;

namespace mindspore {

class TrainAccuracy: public TrainCallBack {
 public:
  explicit TrainAccuracy(int print_every_n = INT_MAX,
                         int accuracy_metrics = METRICS_CLASSIFICATION,
                         const std::vector<int> &input_indexes = {1},
                         const std::vector<int> &output_indexes = {0});
  virtual ~TrainAccuracy();
  const std::vector<GraphPoint> &GetAccuracyPoints();
};
}  // namespace mindspore
#endif  // MINDSPORE_INCLUDE_API_CALLBACK_TRAIN_ACCURACY_H