task_type.h

Go to the documentation of this file.

 
#ifndef TREELITE_ENUM_TASK_TYPE_H_
#define TREELITE_ENUM_TASK_TYPE_H_
 
#include <cstdint>
#include <string>
 
namespace treelite {
 
enum class TaskType : std::uint8_t {
  kBinaryClf = 0,
  kRegressor = 1,
  kMultiClf = 2,
  kLearningToRank = 3,
  kIsolationForest = 4
};
 
std::string TaskTypeToString(TaskType type);
 
TaskType TaskTypeFromString(std::string const& str);
 
}  // namespace treelite
 
#endif  // TREELITE_ENUM_TASK_TYPE_H_