我编写了以下代码,这引发了错误。我想知道我应该修改我的代码以便成功执行以及控制其操作的规则,以便将来避免该问题。
我在这里要做的是定义一个DataLoader类,该类加载语料库,分为训练和测试集,以及完整和测试vocab集。所有这些都没有问题。
当我尝试继承Model类中的DataLoader类时,会出现问题。我想用DataLoader类的所有属性以及一个附加属性n初始化Model类的实例。我不清楚为什么这段代码不能达到这种效果。
from nltk.corpus import brown
class DataLoader():
def __init__(self,data,train_ratio):
self.all = [sent for sent in data.sents()]
num_ex = len(self.all)
num_train = int(train_ratio * num_ex)
self.train = [word.lower() for sent in self.all[:num_train] for word in sent]
self.test = [word.lower() for sent in self.all[num_train:] for word in sent]
self.full_vocab = [word.lower() for word in set(data.words())]
self.train_vocab = list(set(self.train))
return None
class Model(DataLoader):
def __init__(self,data,train_ratio,n):
self.n = n
super().__init__(self,data,train_ratio)
bgrams = Model(data=brown, train_ratio=0.5, n=2)
出现以下错误:
---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
<ipython-input-16-d641225c06d6> in <module>()
----> 1 bgrams = Model(data=brown,train_ratio=0.5,n=2)
2 # bgrams.compute_bigram()
<ipython-input-15-98b0485bce38> in __init__(self, data, train_ratio, n)
9
10 self.n = n
---> 11 super(Model).__init__(self,data,train_ratio)
12
13
TypeError: super() takes at most 2 arguments (3 given)
答案 0 :(得分:2)
简单,只需从super子句中删除self;您的代码应如下所示:
class Model(DataLoader):
def __init__(self,data,train_ratio,n):
self.n = n
super().__init__(data,train_ratio)
答案 1 :(得分:1)
您不必传递self
,按照惯例,super
调用是第一个调用,因为它会影响下一个初始化语句
class Model(DataLoader):
def __init__(self, data, train_ratio, n):
super().__init__(data, train_ratio)
self.n = n