linly/Mining-Away
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

migrated naive based stuff to notebook

Browse Source
This commit is contained in:
LinlyBoi
2023-05-15 19:11:26 +03:00
parent 957451ae33
commit 65d268a902
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
dwarves/mining_hq.py
View File

@@ -8,6 +8,7 @@ import scipy as scp
from sklearn.model_selection import train_test_split from sklearn.model_selection import train_test_split
from sklearn.cluster import KMeans from sklearn.cluster import KMeans
from sklearn import metrics from sklearn import metrics
from sklearn.naive_bayes import GaussianNB
import gunner import gunner
import digger import digger
import scout import scout
@@ -145,3 +146,7 @@ silh_score = metrics.silhouette_score(
print(silh_score) print(silh_score)
gammas_train["Kmean Labels"] = gammas_labels gammas_train["Kmean Labels"] = gammas_labels
print(gammas_train.head()) print(gammas_train.head())
# Naive based
gnb = GaussianNB()
prediction = gnb.fit(gammas_train)