Naive Bayes Klassifizierung in Python

Question

Ich bin ziemlich neu in Python. Ich habe alle Daten aus CSV-Datei

import csv 
import nltk 

f = open('C:/Users/Documents/Data/exp.csv') 
csv_f = csv.reader(f) 

dataset = [] 

for row in csv_f: 
    dataset.append(row) 

print (dataset) 

Jetzt mit lesen, ich tun möchte nltk.NaiveBayesClassifier Wie kann ich das tun?

Answer 1

Zum Beispiel, wenn der Inhalt der CSV ist die folgende:

CSV

Size,Color,Shape,Accept 
small,blue,oval,yes 
small,green,oval,yes 
big,green,oval,no 
big,red,square,no 
small,red,square,no 
small,blue,square,yes 
big,red,circle,yes 

Und wir wollen wissen, ob ein kleine rot-oval Artikel sein akzeptiert mit dem nltk Naive Bayes, können wir den folgenden Code verwenden:

python

import csv 
import nltk 

f = open('C:/Users/Amrit/Documents/Data/exp.csv') 
csv_f = csv.reader(f) 
csv_f.next() #skip the header line 

dataset = [] 

for row in csv_f: 
    dataset.append(({'size': row[0], 'color': row[1], 'shape': row[2]}, row[3])) 

print (dataset) 

classifier = nltk.NaiveBayesClassifier.train(dataset) 

mydata = {'size':'small', 'color':'red', 'shape':'oval'} 
print (mydata, classifier.classify(mydata)) 

Hinweis: Ich lerne auch. Danke für die Links von @ Franscisco Couzo und @Milad M