Corrected typo in tutorial feature importance

sylvaticus · May 14, 2024 · 5501e1d · 5501e1d
1 parent c3c79d3
commit 5501e1d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/src/tutorials/Feature importance/Feature_importance.jl b/docs/src/tutorials/Feature importance/Feature_importance.jl
@@ -66,7 +66,7 @@ bar(string.(sortperm(sobol_by_col)),sobol_by_col[sortperm(sobol_by_col)],label="
 
 # For Shapley values we need first to have a trained model
 m = RandomForestEstimator(rng=TEMPRNG)
-fit!(m,xtrain,ytrain)
+fit!(m,x,y)
 
 function predict_function(model, data)
   data_pred = DataFrame(y_pred = BetaML.predict(model, Matrix(data)))