-
Notifications
You must be signed in to change notification settings - Fork 9
/
nn.py
32 lines (27 loc) · 1.13 KB
/
nn.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import numpy as np
class NeuralNetwork:
def __init__(self, layer_sizes):
"""
Neural Network initialization.
Given layer_sizes as an input, you have to design a Fully Connected Neural Network architecture here.
:param layer_sizes: A list containing neuron numbers in each layers. For example [3, 10, 2] means that there are
3 neurons in the input layer, 10 neurons in the hidden layer, and 2 neurons in the output layer.
"""
# TODO (Implement FCNNs architecture here)
pass
def activation(self, x):
"""
The activation function of our neural network, e.g., Sigmoid, ReLU.
:param x: Vector of a layer in our network.
:return: Vector after applying activation function.
"""
# TODO (Implement activation function here)
pass
def forward(self, x):
"""
Receives input vector as a parameter and calculates the output vector based on weights and biases.
:param x: Input vector which is a numpy array.
:return: Output vector
"""
# TODO (Implement forward function here)
pass