修复 Model Trainer 中的 TypeError：缺少位置参数-Python教程-PHP中文网

修复 Model Trainer 中的 TypeError：缺少位置参数

心靈之曲

发布： 2025-10-21 13:00:46

原创

594人浏览过

修复 model trainer 中的 typeerror：缺少位置参数

本文旨在解决在机器学习模型训练过程中遇到的 `TypeError: initiate_model_training() missing 4 required positional arguments` 错误。通过分析错误原因，并结合代码示例，提供详细的解决方案，帮助读者顺利完成模型训练流程。

在进行机器学习项目时，经常会遇到各种各样的错误。其中，TypeError: initiate_model_training() missing 4 required positional arguments: 'X_train', 'X_test', 'y_train', and 'y_test' 是一个常见的错误，尤其是在构建模型训练流水线时。这个错误表明在调用 initiate_model_training() 函数时，缺少了四个必需的位置参数：训练集特征 X_train、测试集特征 X_test、训练集目标变量 y_train 和测试集目标变量 y_test。

错误原因分析

从提供的代码片段可以看出，initiate_model_training() 函数的定义如下：

def initiate_model_training(self, X_train, X_test, y_train, y_test):
    # 函数体

登录后复制

这表明该函数期望接收四个参数，但实际调用时，并没有传递这些参数。

model_trainer_config.initiate_model_training()

登录后复制

解决方案

解决此问题的关键在于确保在调用 initiate_model_training() 函数时，正确地传递了 X_train, X_test, y_train 和 y_test 这四个参数。以下提供两种常见的解决方案：

方案一：直接传递参数

绘蛙AI修图

绘蛙平台AI修图工具，支持手脚修复、商品重绘、AI扩图、AI换色

129

查看详情

这是最直接的解决方案。在调用 initiate_model_training() 函数之前，确保已经加载或生成了 X_train, X_test, y_train 和 y_test 变量，并将它们作为参数传递给函数。

try:
    config = ConfigurationManager()
    model_trainer_config = config.get_model_trainer_config()
    model_trainer = ModelTrainer(model_trainer_config)

    # 加载训练数据和测试数据，并划分特征和目标变量
    train_data = pd.read_csv(model_trainer_config.train_data_path)  # 假设路径已配置
    test_data = pd.read_csv(model_trainer_config.test_data_path)    # 假设路径已配置

    X_train = train_data.drop(model_trainer_config.target_column, axis=1)
    X_test = test_data.drop(model_trainer_config.target_column, axis=1)
    y_train = train_data[model_trainer_config.target_column]
    y_test = test_data[model_trainer_config.target_column]

    model_trainer.initiate_model_training(X_train, X_test, y_train, y_test)

except Exception as e:
    raise e

登录后复制

方案二：在函数内部加载数据

如果 X_train, X_test, y_train 和 y_test 的生成逻辑与 ModelTrainer 类紧密相关，可以将数据加载和划分的逻辑移动到 initiate_model_training() 函数内部。这样，在调用函数时，就不需要显式地传递这些参数。

class ModelTrainer:
    def __init__(self, model_trainer_config):
        self.model_trainer_config = model_trainer_config

    def initiate_model_training(self):
        try:
            # 从配置文件中读取数据路径和目标列名
            train_data_path = self.model_trainer_config.train_data_path
            test_data_path = self.model_trainer_config.test_data_path
            target_column = self.model_trainer_config.target_column

            # 加载训练数据和测试数据
            train_data = pd.read_csv(train_data_path)
            test_data = pd.read_csv(test_data_path)

            # 划分特征和目标变量
            X_train = train_data.drop(target_column, axis=1)
            X_test = test_data.drop(target_column, axis=1)
            y_train = train_data[target_column]
            y_test = test_data[target_column]

            logger.info('Splitting ')

            models={
            'LinearRegression':LinearRegression(),
            'Lasso':Lasso(),
            'Ridge':Ridge(),
            'Elasticnet':ElasticNet(),
            'RandomForestRegressor': RandomForestRegressor(),
            'GradientBoostRegressor()' : GradientBoostingRegressor(),
            "AdaBoost" : AdaBoostRegressor(),
            'DecisionTreeRegressor' : DecisionTreeRegressor(),
            "SupportVectorRegressor" : SVR(),
            "KNN" : KNeighborsRegressor()
            }

            model_report:dict = ModelTrainer.evaluate_model(X_train,y_train, X_test, y_test, models)
            print(model_report)
            print("\n====================================================================================")
            logger.info(f'Model Report : {model_report}')

            # to get best model score from dictionary
            best_model_score = max(sorted(model_report.values()))

            best_model_name = list(model_report.keys())[
                list(model_report.values()).index(best_model_score)
            ]

            best_model = models[best_model_name]

            print(f"Best Model Found, Model Name :{best_model_name}, R2-score: {best_model_score}")
            print("\n====================================================================================")
            logger.info(f"Best Model Found, Model name: {best_model_name}, R2-score: {best_model_score}")
            logger.info(f"{best_model.feature_names_in_}")

            ModelTrainer.save_obj(
            file_path = self.model_trainer_config.trained_model_file_path,
            obj = best_model
            )


        except Exception as e:
            logger.info('Exception occured at model trianing')
            raise e

登录后复制

相应的调用方式也需要修改：

try:
    config = ConfigurationManager()
    model_trainer_config = config.get_model_trainer_config()
    model_trainer = ModelTrainer(model_trainer_config)
    model_trainer.initiate_model_training() # 无需传递参数
except Exception as e:
    raise e

登录后复制

注意事项

配置文件检查： 确保 model_trainer_config 对象包含了正确的数据路径和目标列名等信息。
数据加载： 确保数据加载过程没有错误，并且数据格式与模型的要求一致。
参数传递： 仔细检查参数传递的顺序和类型，确保与函数定义一致。
代码调试： 使用调试器可以帮助你更清晰地了解代码的执行流程，并找到错误的原因。

总结

TypeError: initiate_model_training() missing 4 required positional arguments 错误通常是由于在调用函数时缺少必要的参数造成的。通过仔细分析错误信息和代码，选择合适的解决方案，并注意相关的事项，可以有效地解决这个问题，顺利完成模型训练任务。选择哪种方案取决于具体的项目结构和数据处理流程。如果数据加载和划分逻辑相对独立，方案一更合适；如果与 ModelTrainer 类紧密相关，方案二更简洁。

以上就是修复 Model Trainer 中的 TypeError：缺少位置参数的详细内容，更多请关注php中文网其它相关文章！