Comment spécifier la probabilité a priori pour scikit-learn Naive Bayes

Je suis en utilisant le scikit-learn apprentissage de la machine de la bibliothèque (Python) pour une machine d'apprentissage du projet. L'un des algorithmes que j'utilise est la Gaussienne Naive Bayes mise en œuvre. L'un des attributs de la GaussianNB() fonction est la suivante:

class_prior_ : array, shape (n_classes,)

Je veux modifier la classe avant manuellement, car les données que j'utilise est très asymétrique et le rappel de l'une des classes est très important. Par l'attribution d'une haute probabilité a priori de la classe le rappel devrait augmenter.

Cependant, je ne peux pas comprendre comment définir l'attribut correctement. J'ai lu ci-dessous les sujets déjà mais leurs réponses ne fonctionnent pas pour moi.

Comment les probabilités antérieures définir manuellement pour la Naive Bayes fcf dans scikit-learn?

Comment puis-je savoir ce que avant je suis en train de donner à la sci-kit d'apprendre? (Naïve-bayes classificateurs.)

C'est mon code:

gnb = GaussianNB()
gnb.class_prior_ = [0.1, 0.9]
gnb.fit(data.XTrain, yTrain)
yPredicted = gnb.predict(data.XTest)

J'ai pensé que c'était la syntaxe correcte et je pouvais savoir à quelle classe appartient à qui la place dans le tableau de jouer avec les valeurs mais les résultats restent inchangés. Aussi aucune erreur n'a été donnée.

Quelle est la façon correcte de la définition des attributs de la GaussianNB algorithme de scikit-learn bibliothèque?

Lien vers le scikit documentation de GaussianNB

OriginalL'auteur pevadi | 2015-06-17