从另一个类内部的类调用函数

Question

我有类配置内部类预处理器，我想调用 get_data 函数

from torch_geometric.datasets import Planetoid
from torch_geometric.data import DataLoader
class Preprocess():
    class config():    
      root_dir = './data/'
      batch_size = 1
      hidden_dim = 16 
      dropout_rate = 0.5
      l2_reg = 5*10e-4
      learning_rate = 0.01
      epochs = 200
      loss_type = 'cross_entropy' # or nll_loss

  def __init__(self, config, d_name):
    self.root_dir = config.root_dir
    self.batch_size = config.batch_size
    
    self.cora = Planetoid(root='./data/cora', name='Cora')
    #self.citeseer = Planetoid(root='./data/citeseer', name='CiteSeer')
    #self.pubmed = Planetoid(root='./data/pubmed', name='PubMed')
    
    self.num_classes, self.num_node_features, self.data = self.get_data(d_name)
    
 def get_data(self, d_name):
    '''
        d_name = 'Cora', 'CiteSeer', 'PubMed'
    '''
    dataset = Planetoid(root=self.root_dir + d_name , name=d_name)
    
    return dataset.num_classes, dataset.num_node_features, DataLoader(dataset, batch_size = self.batch_size)
f=Preprocess(config, Cora)
f.get_data(Cora)

但我有错误： NameError: name 'config' is not defined

Answer 1

您需要使用 Preprocess.config 或 self.config 来访问配置类

Answer 2

你需要把 Preprocess.config 因为那是它定义的地方