13 lines
497 B
Plaintext
13 lines
497 B
Plaintext
from sklearn.linear_model import LogisticRegression
|
|
from sklearn.naive_bayes import MultinomialNB
|
|
from sklearn.multioutput import MultiOutputClassifier
|
|
|
|
#LogisticRegression(max_iter=1337, random_state=0) -> bad macros
|
|
base_clf = MultinomialNB(alpha=0.25)
|
|
|
|
# n_jobs=1 since there seems to be some multithreading join issue in sklearn (or my pc is to bad)
|
|
multi_target_clf = MultiOutputClassifier(base_clf, n_jobs=1)
|
|
|
|
multi_target_clf.fit(X_train, y_train)
|
|
|
|
y_pred = multi_target_clf.predict(X_test) |