Reset cached feature names when loading model

ankane · Nov 12, 2024 · 811bfb4 · 811bfb4
1 parent edaa61a
commit 811bfb4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lib/lightgbm/booster.rb b/lib/lightgbm/booster.rb
@@ -99,6 +99,7 @@ def feature_name
     def model_from_string(model_str)
       out_num_iterations = ::FFI::MemoryPointer.new(:int)
       check_result FFI.LGBM_BoosterLoadModelFromString(model_str, out_num_iterations, @handle)
+      @cached_feature_names = nil
       self
     end