def __init__()

in vision_charts/models.py [0:0]

• 8 lines of code
• 3 McCabe index (conditional complexity)

	def __init__(self, input_features, args):
		super(GCN, self).__init__()

		self.num_layers = args.num_gcn_layers
		# define output sizes for each GCN layer
		hidden_values =  [input_features] + [ args.hidden_gcn_layers for k in range(self.num_layers -1)] + [3]

		# define layers
		layers = []
		for i in range(self.num_layers):
			layers.append(GCN_layer(hidden_values[i], hidden_values[i+1]))
		self.layers = nn.ModuleList(layers)