def test_mapping(): mlp = MLP() w, b = mlp._layermapping(n_features=10, n_classes=11, hidden_layers=[8, 7, 6]) expect_b = {1: [[8], 'n_hidden_1'], 2: [[7], 'n_hidden_2'], 3: [[6], 'n_hidden_3'], 'out': [[11], 'n_classes']} expect_w = {1: [[10, 8], 'n_features, n_hidden_1'], 2: [[8, 7], 'n_hidden_1, n_hidden_2'], 3: [[7, 6], 'n_hidden_2, n_hidden_3'], 'out': [[6, 11], 'n_hidden_3, n_classes']} assert expect_b == b, b assert expect_w == w, w
def test_mapping(): mlp = MLP() w, b = mlp._layermapping(n_features=10, n_classes=11, hidden_layers=[8, 7, 6]) expect_b = { "1": [[8], "n_hidden_1"], "2": [[7], "n_hidden_2"], "3": [[6], "n_hidden_3"], "out": [[11], "n_classes"], } expect_w = { "1": [[10, 8], "n_features, n_hidden_1"], "2": [[8, 7], "n_hidden_1, n_hidden_2"], "3": [[7, 6], "n_hidden_2, n_hidden_3"], "out": [[6, 11], "n_hidden_3, n_classes"], } assert expect_b == b, b assert expect_w == w, w