Program Listing for File train_accuracy.h

↰ Return to documentation for file (include/runtime/include/api/callback/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"

namespace mindspore {

class MS_API 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