def main(): # train_X = np.asarray([[0.2, -0.3]]) # train_Y = np.asarray([[0.0, 1.0, 0.0]]) train_X = np.asarray([[0.2, -0.3], [0.1, -0.9]]) train_Y = np.asarray([[0.0, 1.0, 0.0], [0.0, 0.0, 1.0]]) net = RelativeNetwork(cost="mse", relative_input=RelativeInput(2)) net.append(Relative(2, 5, activation="sigmoid")) net.append(Relative(5, 3, activation="sigmoid")) print(net.predict(train_X)) print(net.cost(train_X, train_Y))