我试图使用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'?如果是,我如何避免这个问题?
我发现问题出在哪里了。
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)
我没有正确继承超类的属性。