add more tests

JuliaAI · Mar 22, 2024 · b9d8c01 · b9d8c01
1 parent 0e59f32
commit b9d8c01
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/test/tuned_models.jl b/test/tuned_models.jl
@@ -157,6 +157,16 @@ end
 
 @testset_accelerated "Feature Importances" accel begin
     # the DecisionTreeClassifier in /test/_models/ supports feature importances.
+    tm0  = TunedModel(                                                                                         
+        model = trees[1],                                                                                                   
+        measure = rms,                                                                                                      
+        tuning = Grid(),                                                                                             
+        resampling = CV(nfolds = 5),                                                                              
+        range = range(                                                                                                      
+            trees[1], :max_depth, values = 1:10                                                                             
+        )                                                                                                           
+    )
+    @test reports_feature_importances(typeof(tm0))
     tm = TunedModel(
         models = trees,
         resampling = CV(nfolds=2),