1

機能 (人の名前) とラベル (人の民族) を含む csv ファイルがあります。Python Pandas を使用してデータ フレームをセットアップできますが、それを NLTK モジュールとリンクしてナイーブ ベイズを実行しようとすると、次のエラーが発生します。

Traceback (most recent call last):
  File "C:\Users\Desktop\file.py", line 19, in <module>
classifier = nbc.train(train_set)
  File "E:\Program Files Extra\Python27\lib\site-packages\nltk\classify\naivebayes.py", line 194, in train
for fname, fval in featureset.items():
AttributeError: 'str' object has no attribute 'items'

これが私のコードです：

import pandas as pd
from pandas import DataFrame
import re
import numpy as np
import nltk
from nltk.classify import NaiveBayesClassifier as nbc

data = pd.read_csv("C:\Users\KubiK\Desktop\OddNames_sampleData3.csv")
frame = DataFrame(data)
frame.columns = ["feature", "label"]
feature = frame.feature
label = frame.label

# Extract features. 
featuresets = [(feature, label) for index, (feature, label) in frame.iterrows()]
# Split train and test set
train_set, test_set = featuresets[:400], featuresets[400:]
# Train a classifier
classifier = nbc.train(train_set)
# Test classifier on "Neo"
print classifier.classify(ethnic_features('Silva'))

サンプルデータ：

Name    Ethnicity
J-b'te Letourneau   Scotish
Jane Mc-earthar French
Li Chen Chinese
Amabil?? Bonneau    English
Emma Lef??c French
C., Akeefe  African
D, James Matheson   English
4

1 に答える 1

1

この行

featuresets = [(feature, label) for index, (feature, label) in frame.iterrows()]

窒息している nbc.train()

featuresets は [(featureset, label)] の形式である必要があります。ここで、featureset 変数はdict (str ではありません) であり、label は featureset の既知のクラス ラベルです。

だからそうあるべきだ

featuresets = [(ethnic_features(feature), label) for index, (feature, label) in frame.iterrows()]

あなたのスニペットにはethnic_features()が含まれていませんでしたが、辞書が返されることを願っています。

于 2015-03-30T07:35:17.593 に答える