Skip to content

Commit c4f1ad3

Browse files
committed
removed flux from tested models because it requires too much test time
1 parent 371412b commit c4f1ad3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/conformal_models/conformal_models.jl

+2-2
Original file line numberDiff line numberDiff line change
@@ -113,14 +113,14 @@ const tested_atomic_models = Dict(
113113
:evo_tree => :(@load EvoTreeRegressor pkg = EvoTrees),
114114
:nearest_neighbor => :(@load KNNRegressor pkg = NearestNeighborModels),
115115
:light_gbm => :(@load LGBMRegressor pkg = LightGBM),
116-
:neural_network => :(@load NeuralNetworkRegressor pkg = MLJFlux),
116+
# :neural_network => :(@load NeuralNetworkRegressor pkg = MLJFlux),
117117
),
118118
:classification => Dict(
119119
:logistic => :(@load LogisticClassifier pkg = MLJLinearModels),
120120
:evo_tree => :(@load EvoTreeClassifier pkg = EvoTrees),
121121
:nearest_neighbor => :(@load KNNClassifier pkg = NearestNeighborModels),
122122
:light_gbm => :(@load LGBMClassifier pkg = LightGBM),
123-
:neural_network => :(@load NeuralNetworkClassifier pkg = MLJFlux),
123+
# :neural_network => :(@load NeuralNetworkClassifier pkg = MLJFlux),
124124
),
125125
)
126126

0 commit comments

Comments
 (0)