85b2fb3eee
* Integrating a faster version of grow_gpu plugin 1. Removed the older files to reduce duplication 2. Moved all of the grow_gpu files under 'exact' folder 3. All of them are inside 'exact' namespace to avoid any conflicts 4. Fixed a bug in benchmark.py while running only 'grow_gpu' plugin 5. Added cub and googletest submodules to ease integration and unit-testing 6. Updates to CMakeLists.txt to directly build cuda objects into libxgboost * Added support for building gpu plugins through make flow 1. updated makefile and config.mk to add right targets 2. added unit-tests for gpu exact plugin code * 1. Added support for building gpu plugin using 'make' flow as well 2. Updated instructions for building and testing gpu plugin * Fix travis-ci errors for PR#2360 1. lint errors on unit-tests 2. removed googletest, instead depended upon dmlc-core provide gtest cache * Some more fixes to travis-ci lint failures PR#2360 * Added Rory's copyrights to the files containing code from both. * updated copyright statement as per Rory's request * moved the static datasets into a script to generate them at runtime * 1. memory usage print when silent=0 2. tests/ and test/ folder organization 3. removal of the dependency of googletest for just building xgboost 4. coding style updates for .cuh as well * Fixes for compilation warnings * add cuda object files as well when JVM_BINDINGS=ON
93 lines
2.5 KiB
C++
93 lines
2.5 KiB
C++
/*!
|
|
* Copyright 2014 by Contributors
|
|
* \file feature_map.h
|
|
* \brief Feature map data structure to help visualization and model dump.
|
|
* \author Tianqi Chen
|
|
*/
|
|
#ifndef XGBOOST_FEATURE_MAP_H_
|
|
#define XGBOOST_FEATURE_MAP_H_
|
|
|
|
#include <vector>
|
|
#include <string>
|
|
#include <cstring>
|
|
#include <iostream>
|
|
|
|
namespace xgboost {
|
|
/*!
|
|
* \brief Feature map data structure to help text model dump.
|
|
* TODO(tqchen) consider make it even more lightweight.
|
|
*/
|
|
class FeatureMap {
|
|
public:
|
|
/*! \brief type of feature maps */
|
|
enum Type {
|
|
kIndicator = 0,
|
|
kQuantitive = 1,
|
|
kInteger = 2,
|
|
kFloat = 3
|
|
};
|
|
/*!
|
|
* \brief load feature map from input stream
|
|
* \param is Input text stream
|
|
*/
|
|
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());
|
|
}
|
|
}
|
|
/*!
|
|
* \brief push back feature map.
|
|
* \param fid The feature index.
|
|
* \param fname The feature name.
|
|
* \param ftype The feature type.
|
|
*/
|
|
inline void PushBack(int fid, const char *fname, const char *ftype) {
|
|
CHECK_EQ(fid, static_cast<int>(names_.size()));
|
|
names_.push_back(std::string(fname));
|
|
types_.push_back(GetType(ftype));
|
|
}
|
|
/*! \brief clear the feature map */
|
|
inline void Clear() {
|
|
names_.clear();
|
|
types_.clear();
|
|
}
|
|
/*! \return number of known features */
|
|
inline size_t size() const {
|
|
return names_.size();
|
|
}
|
|
/*! \return name of specific feature */
|
|
inline const char* name(size_t idx) const {
|
|
CHECK_LT(idx, names_.size()) << "FeatureMap feature index exceed bound";
|
|
return names_[idx].c_str();
|
|
}
|
|
/*! \return type of specific feature */
|
|
Type type(size_t idx) const {
|
|
CHECK_LT(idx, names_.size()) << "FeatureMap feature index exceed bound";
|
|
return types_[idx];
|
|
}
|
|
|
|
private:
|
|
/*!
|
|
* \return feature type enum given name.
|
|
* \param tname The type name.
|
|
* \return The translated type.
|
|
*/
|
|
inline static Type GetType(const char* tname) {
|
|
using namespace std;
|
|
if (!strcmp("i", tname)) return kIndicator;
|
|
if (!strcmp("q", tname)) return kQuantitive;
|
|
if (!strcmp("int", tname)) return kInteger;
|
|
if (!strcmp("float", tname)) return kFloat;
|
|
LOG(FATAL) << "unknown feature type, use i for indicator and q for quantity";
|
|
return kIndicator;
|
|
}
|
|
/*! \brief name of the feature */
|
|
std::vector<std::string> names_;
|
|
/*! \brief type of the feature */
|
|
std::vector<Type> types_;
|
|
};
|
|
} // namespace xgboost
|
|
#endif // XGBOOST_FEATURE_MAP_H_
|