Small changes

JuliaAI · Feb 14, 2023 · 413a2d9 · 413a2d9
1 parent dbcb91d
commit 413a2d9
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/src/DecisionTree.jl b/src/DecisionTree.jl
@@ -5,7 +5,7 @@ using DelimitedFiles
 using LinearAlgebra
 using Random
 using Statistics
-using AbstractTrees: AbstractTrees
+import AbstractTrees
 
 export Leaf,
     Node,

diff --git a/src/classification/main.jl b/src/classification/main.jl
@@ -512,7 +512,7 @@ function build_forest(
                 min_samples_split,
                 min_purity_increase;
                 loss,
-                impurity_importance=impurity_importance,
+                impurity_importance,
             )
         end
     end

diff --git a/src/regression/main.jl b/src/regression/main.jl
@@ -122,7 +122,7 @@ function build_forest(
                 min_samples_split,
                 min_purity_increase;
                 rng=_rng,
-                impurity_importance=impurity_importance,
+                impurity_importance,
             )
         end
     else # each thread gets its own seeded rng

diff --git a/test/regression/energy.jl b/test/regression/energy.jl
@@ -2,11 +2,11 @@
 # https://archive.ics.uci.edu/ml/datasets/Appliances+energy+prediction
 
 @testset "energy.jl" begin
-    download(
-        "https://archive.ics.uci.edu/ml/machine-learning-databases/00374/energydata_complete.csv",
-        "energy.csv",
-    )
-    energy = readcsv("energy.csv")
+    energy = let
+        url = "https://archive.ics.uci.edu/ml/machine-learning-databases/00374/energydata_complete.csv"
+        tmp_path = download(url)
+        readcsv(tmp_path)
+    end
 
     features = energy[2:end, 3:end]
     labels = float.(energy[2:end, 2])