removed flux from tested models because it requires too much test time

JuliaTrustworthyAI · May 19, 2023 · c4f1ad3 · c4f1ad3 · pat-alt · May 20, 2023
1 parent 371412b
commit c4f1ad3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/conformal_models/conformal_models.jl b/src/conformal_models/conformal_models.jl
@@ -113,14 +113,14 @@ const tested_atomic_models = Dict(
         :evo_tree => :(@load EvoTreeRegressor pkg = EvoTrees),
         :nearest_neighbor => :(@load KNNRegressor pkg = NearestNeighborModels),
         :light_gbm => :(@load LGBMRegressor pkg = LightGBM),
-        :neural_network => :(@load NeuralNetworkRegressor pkg = MLJFlux),
+        # :neural_network => :(@load NeuralNetworkRegressor pkg = MLJFlux),
     ),
     :classification => Dict(
         :logistic => :(@load LogisticClassifier pkg = MLJLinearModels),
         :evo_tree => :(@load EvoTreeClassifier pkg = EvoTrees),
         :nearest_neighbor => :(@load KNNClassifier pkg = NearestNeighborModels),
         :light_gbm => :(@load LGBMClassifier pkg = LightGBM),
-        :neural_network => :(@load NeuralNetworkClassifier pkg = MLJFlux),
+        # :neural_network => :(@load NeuralNetworkClassifier pkg = MLJFlux),
     ),
 )