当我试图将一个列表传递给我的类属性时,会发生错误。

问题描述 投票:0回答:1

我试图使用classmethod向我的类属性传递一个列表。

class BASE_MODEL(object):
    def __init__(self, train_data, train_labels, val_data, val_labels):
        self.train_data, self.train_labels, self.val_data, self.val_labels = \
            train_data, train_labels, val_data, val_labels

    @classmethod
    def list_to_data(cls, data_list):
        train_data, train_labels, val_data, val_labels = data_list
        data = cls(train_data, train_labels, val_data, val_labels)
        return data

class Fine_Tune_Model(BASE_MODEL):
    def __init__(self, fine_tune_at=None):
        super(BASE_MODEL, self).__init__()
        self.fine_tune_at = fine_tune_at

fine_tune_model = Fine_Tune_Model.list_to_data(data_lsit)

这对超类来说很好。

但是当我尝试将同样的函数传递给子类时,出现了一个错误。

Traceback (most recent call last):
  File "E:/Studium/Thesis/Fine_Tuning.py", line 138, in <module>
    main()
  File "E:/Studium/Thesis/Fine_Tuning.py", line 133, in main
    fine_tune_model = Fine_Tune_Model.list_to_data(create_data.split_data(3000, 200))
  File "E:\Studium\Thesis\Train.py", line 32, in list_to_data
    data = cls(train_data, train_labels, val_data, val_labels)
TypeError: __init__() takes from 1 to 2 positional arguments but 5 were given

Process finished with exit code 1

是不是因为我的子类中多了一个参数'fine_tune_at'?如果是,我如何避免这个问题?

python class inheritance subclass superclass
1个回答
0
投票

我发现问题出在哪里了。

class BASE_MODEL(object):
    def __init__(self, train_data, train_labels, val_data, val_labels):
        self.train_data, self.train_labels, self.val_data, self.val_labels = \
            train_data, train_labels, val_data, val_labels

    @classmethod
    def list_to_data(cls, data_list):
        train_data, train_labels, val_data, val_labels = data_list
        data = cls(train_data, train_labels, val_data, val_labels)
        return data

class Fine_Tune_Model(BASE_MODEL):
    def __init__(self, train_data, train_labels, val_data, val_labels):
        super(Fine_Tune_Model, self).__init__(train_data, train_labels, val_data, val_labels)
        self.fine_tune_at = fine_tune_at

fine_tune_model = Fine_Tune_Model.list_to_data(data_lsit)

我没有正确继承超类的属性。

© www.soinside.com 2019 - 2024. All rights reserved.