Skip to content

Commit f4ee3fe

Browse files
minor update to examples (#206)
1 parent 5b26035 commit f4ee3fe

File tree

2 files changed

+25
-7
lines changed

2 files changed

+25
-7
lines changed

examples/logistic_regression.py

+11-2
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,27 @@
1-
from nums import numpy as nps
1+
import nums
2+
import nums.numpy as nps
23
from nums.models.glms import LogisticRegression
34

5+
nums.init()
6+
47

58
# Make dataset.
9+
610
X1 = nps.random.randn(500, 1) + 5.0
711
y1 = nps.zeros(shape=(500,), dtype=bool)
12+
813
X2 = nps.random.randn(500, 1) + 10.0
914
y2 = nps.ones(shape=(500,), dtype=bool)
15+
1016
X = nps.concatenate([X1, X2], axis=0)
1117
y = nps.concatenate([y1, y2], axis=0)
1218

1319

1420
# Train Logistic Regression Model.
15-
model = LogisticRegression(solver="newton-cg", tol=1e-8, max_iter=1)
21+
22+
model = LogisticRegression(solver="newton", tol=1e-8, max_iter=1)
23+
1624
model.fit(X, y)
1725
y_pred = model.predict(X)
26+
1827
print("accuracy", (nps.sum(y == y_pred) / X.shape[0]).get())

examples/notebooks/logistic_regression.ipynb

+14-5
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,11 @@
66
"metadata": {},
77
"outputs": [],
88
"source": [
9-
"from nums import numpy as nps\n",
10-
"from nums.models.glms import LogisticRegression"
9+
"import nums\n",
10+
"import nums.numpy as nps\n",
11+
"from nums.models.glms import LogisticRegression\n",
12+
"\n",
13+
"nums.init()"
1114
]
1215
},
1316
{
@@ -17,10 +20,13 @@
1720
"outputs": [],
1821
"source": [
1922
"# Make dataset.\n",
23+
"\n",
2024
"X1 = nps.random.randn(500, 1) + 5.0\n",
2125
"y1 = nps.zeros(shape=(500,), dtype=bool)\n",
26+
"\n",
2227
"X2 = nps.random.randn(500, 1) + 10.0\n",
2328
"y2 = nps.ones(shape=(500,), dtype=bool)\n",
29+
"\n",
2430
"X = nps.concatenate([X1, X2], axis=0)\n",
2531
"y = nps.concatenate([y1, y2], axis=0)"
2632
]
@@ -32,16 +38,19 @@
3238
"outputs": [],
3339
"source": [
3440
"# Train Logistic Regression Model.\n",
35-
"model = LogisticRegression(solver=\"newton-cg\", tol=1e-8, max_iter=1)\n",
41+
"\n",
42+
"model = LogisticRegression(solver=\"newton\", tol=1e-8, max_iter=1)\n",
43+
"\n",
3644
"model.fit(X, y)\n",
3745
"y_pred = model.predict(X)\n",
46+
"\n",
3847
"print(\"accuracy\", (nps.sum(y == y_pred) / X.shape[0]).get())"
3948
]
4049
}
4150
],
4251
"metadata": {
4352
"kernelspec": {
44-
"display_name": "Python 3",
53+
"display_name": "Python 3 (ipykernel)",
4554
"language": "python",
4655
"name": "python3"
4756
},
@@ -55,7 +64,7 @@
5564
"name": "python",
5665
"nbconvert_exporter": "python",
5766
"pygments_lexer": "ipython3",
58-
"version": "3.7.4"
67+
"version": "3.7.10"
5968
}
6069
},
6170
"nbformat": 4,

0 commit comments

Comments
 (0)