From 6b7a4ab005c3e0d79c799587f057aecd96a21795 Mon Sep 17 00:00:00 2001 From: Erik Ordentlich Date: Fri, 9 Jan 2026 16:01:05 -0800 Subject: [PATCH 1/2] fix slicing of coefficient arrays in some notebooks - slicing not supported when these vectors happen to be sparse, mainly cpu case Signed-off-by: Erik Ordentlich --- notebooks/linear-regression.ipynb | 4 ++-- notebooks/logistic-regression.ipynb | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/notebooks/linear-regression.ipynb b/notebooks/linear-regression.ipynb index 86b1e08c..6deee966 100644 --- a/notebooks/linear-regression.ipynb +++ b/notebooks/linear-regression.ipynb @@ -336,7 +336,7 @@ }, "outputs": [], "source": [ - "gpu_model_loaded.coefficients[0:10]" + "gpu_model_loaded.coefficients.toArray()[0:10]" ] }, { @@ -538,7 +538,7 @@ }, "outputs": [], "source": [ - "cpu_model.coefficients[0:10]" + "cpu_model.coefficients.toArray()[0:10]" ] }, { diff --git a/notebooks/logistic-regression.ipynb b/notebooks/logistic-regression.ipynb index 72cceac5..63693932 100644 --- a/notebooks/logistic-regression.ipynb +++ b/notebooks/logistic-regression.ipynb @@ -259,7 +259,7 @@ }, "outputs": [], "source": [ - "gpu_model.coefficients[0:10] if gpu_model.numClasses <= 2 else gpu_model.coefficientMatrix.toArray()[:,0:10]" + "gpu_model.coefficients.toArray()[0:10] if gpu_model.numClasses <= 2 else gpu_model.coefficientMatrix.toArray()[:,0:10]" ] }, { @@ -335,7 +335,7 @@ }, "outputs": [], "source": [ - "gpu_model_loaded.coefficients[0:10] if gpu_model_loaded.numClasses <= 2 else gpu_model_loaded.coefficientMatrix.toArray()[:,0:10]" + "gpu_model_loaded.coefficients.toArray()[0:10] if gpu_model_loaded.numClasses <= 2 else gpu_model_loaded.coefficientMatrix.toArray()[:,0:10]" ] }, { @@ -565,7 +565,7 @@ }, "outputs": [], "source": [ - "cpu_model.coefficients[0:10] if cpu_model.numClasses <= 2 else cpu_model.coefficientMatrix.toArray()[:,0:10]" + "cpu_model.coefficients.toArray()[0:10] if cpu_model.numClasses <= 2 else cpu_model.coefficientMatrix.toArray()[:,0:10]" ] }, { From 8f85bdfe70687076b743c3f7048c92fb0186390d Mon Sep 17 00:00:00 2001 From: Erik Ordentlich Date: Fri, 9 Jan 2026 16:12:33 -0800 Subject: [PATCH 2/2] one more found by greptile Signed-off-by: Erik Ordentlich --- notebooks/linear-regression.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/notebooks/linear-regression.ipynb b/notebooks/linear-regression.ipynb index 6deee966..55e933bc 100644 --- a/notebooks/linear-regression.ipynb +++ b/notebooks/linear-regression.ipynb @@ -269,7 +269,7 @@ }, "outputs": [], "source": [ - "coefs[0:10]" + "coefs.toArray()[0:10]" ] }, {