diff --git a/FullConnectedNetwork.py b/FullConnectedNetwork.py index f371e60..9ead7da 100644 --- a/FullConnectedNetwork.py +++ b/FullConnectedNetwork.py @@ -64,9 +64,7 @@ class Layer(object): def __init__(self,layer_index,node_count): self.layer_index = layer_index self.node_count = node_count - self.nodes = [] - for i in range(self.node_count): - self.nodes.append(Node(layer_index,i)) + self.nodes = [Node(layer_index,i) for i in range(self.node_count)] self.nodes.append(ConstNode(layer_index,self.node_count)) def set_output(self,data):#input layer x1,x2,x3... @@ -125,8 +123,7 @@ def __init__(self,layers):#layers[i] <----> # of nodes in layer i layer_count = len(layers) node_count = 0 #how many layers? - for i in range(layer_count): - self.layers.append(Layer(i,layers[i])) + self.layers.extend(Layer(i,layers[i]) for i in range(layer_count)) for layer in range(layer_count - 1): connections = [Connection(upstream_node,downstream_node) for upstream_node in self.layers[layer].nodes diff --git a/Perceptron.py b/Perceptron.py index e27703f..2ef8be8 100644 --- a/Perceptron.py +++ b/Perceptron.py @@ -15,7 +15,7 @@ def predict(self,input_vec): list(map(lambda x:x[0] * x[1],list(zip(input_vec,self.weights))))) + self.bias) def train(self, input_vec ,labels, iteration,rate): - for i in range(iteration): + for _ in range(iteration): self._one_iteration(input_vec,labels,rate) def _one_iteration(self,input_vec,labels,rate):