Skip to content

Commit 9f77e66

Browse files
committed
feat: add SVM
1 parent 203ced8 commit 9f77e66

File tree

5 files changed

+246
-186
lines changed

5 files changed

+246
-186
lines changed

examples/tools-usages/decomposition-manifold/pca_tsne_kmeans_dbscan.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
"\n",
1313
"from neuralnetlib.preprocessing import StandardScaler\n",
1414
"from neuralnetlib.preprocessing import PCA, TSNE\n",
15-
"from neuralnetlib.cluster import KMeans, DBSCAN\n",
15+
"from neuralnetlib.learners import KMeans, DBSCAN\n",
1616
"\n",
1717
"from sklearn.metrics import adjusted_rand_score, adjusted_mutual_info_score\n",
1818
"from sklearn.datasets import make_blobs"
@@ -249,7 +249,7 @@
249249
"print(f\"Adjusted Mutual Information Score: {adjusted_mutual_info_score(y, y_pred):.3f}\")\n",
250250
"\n",
251251
"print(\"\\nCluster Centers (after de-standardization):\")\n",
252-
"cluster_centers = scaler.inverse_transform(kmeans.cluster_centers_)\n",
252+
"cluster_centers = scaler.inverse_transform(kmeans.learners_centers_)\n",
253253
"for i, center in enumerate(cluster_centers):\n",
254254
" print(f\"Cluster {i}: {center}\")"
255255
]

examples/tools-usages/unsupervised/unsupervised.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
"metadata": {},
77
"outputs": [],
88
"source": [
9-
"from neuralnetlib.ensemble import IsolationForest, RandomForest, AdaBoost, GradientBoostingMachine, XGBoost\n",
9+
"from neuralnetlib.learners import IsolationForest, RandomForest, AdaBoost, GradientBoostingMachine, XGBoost\n",
1010
"\n",
1111
"import numpy as np\n",
1212
"import matplotlib.pyplot as plt\n",

neuralnetlib/__init__.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,5 +8,4 @@
88
from . import preprocessing
99
from . import regularizers
1010
from . import utils
11-
from . import ensemble
12-
from . import cluster
11+
from . import learners

neuralnetlib/cluster.py

Lines changed: 0 additions & 181 deletions
This file was deleted.

0 commit comments

Comments
 (0)