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_gbmbase.h
+++ b/booster/xgboost_gbmbase.h
@@ -188,11 +188,13 @@ namespace xgboost{
            /*! 
             * \brief DumpModel
             * \param fo text file 
+             * \param fmap feature map that may help give interpretations of feature
+             * \param with_stats whether print statistics
             */            
-            inline void DumpModel( FILE *fo ){
+            inline void DumpModel( FILE *fo, const utils::FeatMap& fmap, bool with_stats ){
                for( size_t i = 0; i < boosters.size(); i ++ ){
                    fprintf( fo, "booster[%d]\n", (int)i );
-                    boosters[i]->DumpModel( fo );
+                    boosters[i]->DumpModel( fo, fmap, with_stats );
                }
            }
            /*!