diff --git a/notebooks/linear-regression.ipynb b/notebooks/linear-regression.ipynb index 86b1e08c..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]" ] }, { @@ -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]" ] }, {