2列のcsvを2変数のdefに入れられない

Question

下記のbayes_test.pyを実行すると、以下のようなエラーが出ており、
必要な引数は2つに対し、csvを2列で取り込んでいるのでうまくいくはずのところを、
「カテゴリーという変数が足りない」と表示されます。

csvの取り込みのプロセスで、基礎的なミスがあるのでしょうか？
ご返答いただけましたら幸いです。
よろしくお願いいたします。

Traceback (most recent call last):
File "bayes_test.py", line 11, in <module>
bf.fit(line)
TypeError: fit() missing 1 required positional argument: 'category'

**bayes_test.py**
from bayes import BayesianFilter
import csv

bf = BayesianFilter()
# テキストを学習
data = []
with open('format.csv', 'r', encoding='utf-8') as fp:
    lines = csv.reader(fp)
    for line in lines:
        data.append(line)
        bf.fit(line)
pre, scorelist = bf.predict("点検")
print("結果=", pre)
print(scorelist)

**bayes.py**
# テキストを学習する --- (※3)
    def fit(self, text, category):
        """ テキストの学習 """
        word_list = self.split(text)
        for word in word_list:
            self.inc_word(word, category)
        self.inc_category(category)

Answer 1

エラーメッセージが示す通り、fit関数の引数はtextとcategoryの２つであるのにlineというリスト変数を１つしか渡していないのが原因です。
lineがリストとして２つのデータが格納されていれば、bf.fit(line[0],line[1])と呼び出せばよいです。

Comments

[TarouKotani]

bf.fit(line[0],line[1])に書き換えると下記のエラーになりました
csvが2列であることは確認できているのですが、どのような問題がありそうでしょうか
Traceback (most recent call last):
File "bayes_test.py", line 11, in <module>
bf.fit(line[0],line[1])
IndexError: list index out of range

[can110]

まずエラー「IndexError: list index out of range」の意味はご理解できていますか？

[TarouKotani]

解決しました！ありがとうございます。

[TarouKotani]

下記の通りに直しました。その後に別のエラーが出ましたが、この質問は解決しています。
lines = csv.reader(fp)
for line in lines:
data.append(line)
print(len(line))
bf.fit(line[0],line[1])