fix fmap

2014-03-03 11:05:10 -08:00
parent 074a861e7b
commit 623e003923
11 changed files with 172 additions and 34 deletions
--- a/booster/xgboost.h
+++ b/booster/xgboost.h
@@ -8,6 +8,7 @@
 */
 #include <vector>
 #include "../utils/xgboost_utils.h"
+#include "../utils/xgboost_fmap.h"
 #include "../utils/xgboost_stream.h"
 #include "../utils/xgboost_config.h"
 #include "xgboost_data.h"
@@ -107,8 +108,10 @@ namespace xgboost{
            /*! 
             * \brief dump model into text file
             * \param fo output stream 
-             */        
-            virtual void DumpModel( FILE *fo ){
+             * \param fmap feature map that may help give interpretations of feature
+             * \param with_stats whether print statistics
+             */
+            virtual void DumpModel( FILE *fo, const utils::FeatMap& fmap, bool with_stats = false ){
                utils::Error( "not implemented" );                
            }
        public: