nuluh/thesis
1
0
Fork 0
Code Issues 47 Pull Requests Actions Packages Projects Releases 1 Wiki Activity

[FEAT] Feat Include Undamaged Node Classification #98

Merged
nuluh merged 15 commits from feat/53-feat-include-undamaged-node-classification into dev 2025-06-18 02:06:04 +00:00
Conversation 0 Commits 15 Files Changed 7 +15
1 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 18892c1188 - Show all commits

15
code/notebooks/stft.ipynb
View File

@@ -690,6 +690,10 @@
"source": [ "source": [
"from src.ml.model_selection import train_and_evaluate_model\n", "from src.ml.model_selection import train_and_evaluate_model\n",
"from sklearn.svm import SVC\n", "from sklearn.svm import SVC\n",
"from sklearn.pipeline import make_pipeline\n",
"from sklearn.preprocessing import StandardScaler\n",
"from sklearn.svm import SVC\n",
"from sklearn.decomposition import PCA\n",
"# Define models for sensor1\n", "# Define models for sensor1\n",
"models_sensor1 = {\n", "models_sensor1 = {\n",
" # \"Random Forest\": RandomForestClassifier(),\n", " # \"Random Forest\": RandomForestClassifier(),\n",
@@ -698,6 +702,12 @@
" # \"KNN\": KNeighborsClassifier(),\n", " # \"KNN\": KNeighborsClassifier(),\n",
" # \"LDA\": LinearDiscriminantAnalysis(),\n", " # \"LDA\": LinearDiscriminantAnalysis(),\n",
" \"SVM\": SVC(),\n", " \"SVM\": SVC(),\n",
" \"SVM with StandardScaler and PCA\": make_pipeline(\n",
" StandardScaler(),\n",
" PCA(n_components=10),\n",
" SVC(kernel='rbf')\n",
" ),\n",
"\n",
" # \"XGBoost\": XGBClassifier()\n", " # \"XGBoost\": XGBClassifier()\n",
"}\n", "}\n",
"\n", "\n",
@@ -721,6 +731,11 @@
" # \"KNN\": KNeighborsClassifier(),\n", " # \"KNN\": KNeighborsClassifier(),\n",
" # \"LDA\": LinearDiscriminantAnalysis(),\n", " # \"LDA\": LinearDiscriminantAnalysis(),\n",
" \"SVM\": SVC(),\n", " \"SVM\": SVC(),\n",
" \"SVM with StandardScaler and PCA\": make_pipeline(\n",
" StandardScaler(),\n",
" PCA(n_components=10),\n",
" SVC(kernel='rbf')\n",
" ),\n",
" # \"XGBoost\": XGBClassifier()\n", " # \"XGBoost\": XGBClassifier()\n",
"}\n", "}\n",
"\n", "\n",