feature_map.h

前往此文件的文档。

#ifndef XGBOOST_FEATURE_MAP_H_
#define XGBOOST_FEATURE_MAP_H_

#include <xgboost/logging.h>

#include <vector>
#include <string>
#include <cstring>
#include <iostream>

namespace xgboost {
class FeatureMap {
 public
 enum Type {
 kIndicator = 0,
 kQuantitive = 1,
 kInteger = 2,
 kFloat = 3,
 kCategorical = 4
 };
 inline void LoadText(std::istream& is) { // NOLINT(*)
 int fid;
 std::string fname, ftype;
 while (is >> fid >> fname >> ftype) {
 this->PushBack(fid, fname.c_str(), ftype.c_str());
 }
 }
 inline void PushBack(int fid, const char *fname, const char *ftype) {
 CHECK_EQ(fid, static_cast<int>(names_.size()));
 names_.emplace_back(fname);
 types_.push_back(GetType(ftype));
 }
 inline void Clear() {
 names_.clear();
 types_.clear();
 }
 inline size_t Size() const {
 return names_.size();
 }
 inline const char* Name(size_t idx) const {
 CHECK_LT(idx, names_.size()) << "特征映射的特征索引超出界限";
 return names_[idx].c_str();
 }
 Type TypeOf(size_t idx) const {
 CHECK_LT(idx, names_.size()) << "特征映射的特征索引超出界限";
 return types_[idx];
 }

 private
 inline static Type GetType(const char* tname) {
 using std::strcmp;
 if (!strcmp("i", tname)) return kIndicator;
 if (!strcmp("q", tname)) return kQuantitive;
 if (!strcmp("int", tname)) return kInteger;
 if (!strcmp("float", tname)) return kFloat;
 if (!strcmp("c", tname)) return kCategorical;
 LOG(FATAL) << "未知特征类型，请使用 i 表示指示器，q 表示数量";
 return kIndicator;
 }
 std::vector<std::string> names_;
 std::vector<Type> types_;
};
} // namespace xgboost
#endif // XGBOOST_FEATURE_MAP_H_