Python中ChatterBot训练出现NoneType不可迭代错误的解决办法

错误信息 #

Traceback (most recent call last):
  File "c:\Users\Amank\OneDrive\Documents\chatbot\index.py", line 12, in <module>
    trainer.train(cleaned_corpus)
  File "C:\Users\Amank\AppData\Local\Programs\Python\Python311\Lib\site-packages\chatterbot\trainers.py", line 96, in train
    for conversation_count, text in enumerate(conversation):
                                    ^^^^^^^^^^^^^^^^^^^^^^^
TypeError: 'NoneType' object is not iterable

相关代码 #

def train(self, conversation):
    """
    Train the chat bot based on the provided list of
    statements that represents a single conversation.
    """
    previous_statement_text = None
    previous_statement_search_text = ''

    statements_to_create = []

    for conversation_count, text in enumerate():
        if self.show_training_progress:
            utils.print_progress_bar(
                'List Trainer',
                conversation_count + 1, len(conversation)
            )

        statement_search_text = self.chatbot.storage.tagger.get_bigram_pair_string(text)

        statement = self.get_preprocessed_statement(
            Statement(
                text=text,
                search_text=statement_search_text,
                in_response_to=previous_statement_text,
                search_in_response_to=previous_statement_search_text,
                conversation='training'
            )
        )

        previous_statement_text = statement.text
        previous_statement_search_text = statement_search_text

        statements_to_create.append(statement)

    self.chatbot.storage.create_many(statements_to_create)


class ChatterBotCorpusTrainer(Trainer):
"""
Allows the chat bot to be trained using data from the
ChatterBot dialog corpus.
"""

def train(self, *corpus_paths):
    from chatterbot.corpus import load_corpus, list_corpus_files

    data_file_paths = []

    # Get the paths to each file the bot will be trained with
    for corpus_path in corpus_paths:
        data_file_paths.extend(list_corpus_files(corpus_path))

    for corpus, categories, file_path in load_corpus(*data_file_paths):

        statements_to_create = []

        # Train the chat bot with each statement and response pair
        for conversation_count, conversation in enumerate(corpus):

            if self.show_training_progress:
                utils.print_progress_bar(
                    'Training ' + str(os.path.basename(file_path)),
                    conversation_count + 1,
                    len(corpus)
                )

            previous_statement_text = None
            previous_statement_search_text = ''

            for text in conversation:

                statement_search_text = self.chatbot.storage.tagger.get_bigram_pair_string(text)

                statement = Statement(
                    text=text,
                    search_text=statement_search_text,
                    in_response_to=previous_statement_text,
                    search_in_response_to=previous_statement_search_text,
                    conversation='training'
                )

                statement.add_tags(*categories)

                statement = self.get_preprocessed_statement(statement)

                previous_statement_text = statement.text
                previous_statement_search_text = statement_search_text

                statements_to_create.append(statement)

        self.chatbot.storage.create_many(statements_to_create)

问题排查与修复方案 #

1. 修复自定义train方法中的enumerate参数缺失问题 #

你提供的第一个train函数存在语法错误：for conversation_count, text in enumerate(): 这里enumerate()未传入可迭代对象，根据函数定义，应该遍历传入的conversation参数。这个错误会直接导致遍历失败，若传入的conversation本身为None（比如cleaned_corpus是None），则会触发报错。

修改后的代码：

def train(self, conversation):
    """
    Train the chat bot based on the provided list of
    statements that represents a single conversation.
    """
    previous_statement_text = None
    previous_statement_search_text = ''

    statements_to_create = []

    # 修复：给enumerate传入conversation参数
    for conversation_count, text in enumerate(conversation):
        if self.show_training_progress:
            utils.print_progress_bar(
                'List Trainer',
                conversation_count + 1, len(conversation)
            )

        statement_search_text = self.chatbot.storage.tagger.get_bigram_pair_string(text)

        statement = self.get_preprocessed_statement(
            Statement(
                text=text,
                search_text=statement_search_text,
                in_response_to=previous_statement_text,
                search_in_response_to=previous_statement_search_text,
                conversation='training'
            )
        )

        previous_statement_text = statement.text
        previous_statement_search_text = statement_search_text

        statements_to_create.append(statement)

    self.chatbot.storage.create_many(statements_to_create)

2. 验证cleaned_corpus的有效性 #

检查调用trainer.train(cleaned_corpus)时传入的cleaned_corpus是否为None或空对象：

• 在调用train方法前添加打印语句：print(cleaned_corpus)，确认它是包含对话内容的非空可迭代对象（如列表、元组）。
• 检查数据清洗逻辑，确保cleaned_corpus在生成过程中未被错误赋值为None。

内容的提问来源于stack exchange，提问作者Aman_vdub