Подскажите, пожалуйста, что я не так написал, у меня в переменной "y" хранится много значений (good, maybe, never)
Но когда я вывожу self.labels, то у меня на выходе получается только по одному значению [None, 'good', 'never', 'maybe']
, а нужно чтобы были все good
, все maybe
и never
Вот мой код:
db = sqlite3.connect('news.db')
cur = db.cursor()
X = cur.execute("""SELECT title FROM news""").fetchall()
y = cur.execute("""SELECT label FROM news""").fetchall()
y = [i[0] for i in y]
class NC:
def __init__(self, alpha: int) -> None:
self.alpha = alpha
def fit(self, X, y):
self.labels = list(set(y))
self.class_p = {i: self.labels.count(i) / len(y) for i in self.labels}
a = NC(alpha=1)
a.fit(X, y)
print(a.labels)
print(a.class_p)