fix fmap

2014-03-03 11:05:10 -08:00
parent 8ae1d37828
commit cfbeeef9c1
11 changed files with 172 additions and 34 deletions
--- a/utils/xgboost_fmap.h
+++ b/utils/xgboost_fmap.h
@@ -0,0 +1,68 @@
+#ifndef XGBOOST_FMAP_H
+#define XGBOOST_FMAP_H
+/*!
+ * \file xgboost_fmap.h
+ * \brief helper class that holds the feature names and interpretations
+ * \author Tianqi Chen: tianqi.tchen@gmail.com
+ */
+#include <vector>
+#include <string>
+#include <cstring>
+#include "xgboost_utils.h"
+
+namespace xgboost{
+    namespace utils{
+        /*! \brief helper class that holds the feature names and interpretations */
+        class FeatMap{
+        public:
+            enum Type{
+                kIndicator  = 0,
+                kQuantitive = 1
+            };
+        public:
+            /*! \brief load feature map from text format */
+            inline void LoadText( const char *fname ){
+                FILE *fi = utils::FopenCheck( fname, "r" );
+                this->LoadText( fi );
+                fclose( fi );
+            }
+            /*! \brief load feature map from text format */
+            inline void LoadText( FILE *fi ){
+                int fid;
+                char fname[256], ftype[256];
+                while( fscanf( fi, "%d%s%s", &fid, fname, ftype ) == 3 ){
+                    utils::Assert( fid == (int)names_.size(), "invalid fmap format" );
+                    names_.push_back( std::string(fname) );
+                    types_.push_back( GetType( ftype ) );
+                }
+            }
+            /*! \brief number of known features */
+            size_t size( void ) const{
+                return names_.size();
+            }
+            /*! \brief return name of specific feature */
+            const char* name( size_t idx ) const{
+                utils::Assert( idx < names_.size(), "utils::FMap::name feature index exceed bound" );
+                return names_[ idx ].c_str();
+            }
+            /*! \brief return type of specific feature */
+            const Type& type( size_t idx ) const{
+                utils::Assert( idx < names_.size(), "utils::FMap::name feature index exceed bound" );
+                return types_[ idx ];
+            }
+        private:
+            inline static Type GetType( const char *tname ){
+                if( !strcmp( "i", tname ) ) return kIndicator;        
+                if( !strcmp( "q", tname ) ) return kQuantitive;
+                utils::Error("unknown feature type, use i for indicator and q for quantity");
+                return kIndicator;
+            }
+        private:
+            /*! \brief name of the feature */
+            std::vector<std::string> names_;
+            /*! \brief type of the feature */
+            std::vector<Type>        types_;
+        };
+    }; // namespace utils
+}; // namespace xgboost
+#endif // XGBOOST_FMAP_H