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: 0.45, 1: 3.33}, {1: 2.22}])
train_Y = np.asarray([{1: 1.0}, {2: 1.0}])
net = ParticleSparseNetwork(cost="mse_sparse", particle_input=ParticleSparseInput(2))
net.append(ParticleSparse(2, 3))
# net.append(ParticleSparse(5, 3))
print(net.particle_input.get_rxyz())
print(net.predict(train_X))
print(net.cost(train_X, train_Y))
print(net.cost_gradient(train_X, train_Y))
