linear_regression_intro WIP

scicloj · Dec 11, 2024 · f89fff0 · f89fff0
1 parent 59aa280
commit f89fff0
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/notebooks/noj_book/linear_regression_intro.clj b/notebooks/noj_book/linear_regression_intro.clj
@@ -121,11 +121,13 @@ totals-with-one-hot-days-of-week
 ;; bike count for the corresponding day of week.
 
 (def fit
-  (reg/lm (:total totals-with-one-hot-days-of-week)
-          (-> totals-with-one-hot-days-of-week
-              (tc/drop-columns [:total])
-              tc/rows)
-          {:intercept? false}))
+  (let [data-without-target (-> totals-with-one-hot-days-of-week
+                                (tc/drop-columns [:total]))]
+    data-without-target
+    (reg/lm (:total totals-with-one-hot-days-of-week)
+            (tc/rows data-without-target)
+            {:intercept? false
+             :names (vec (tc/column-names data-without-target))})))
 
 ;; Here are the regression results: