def main():
i_gen = InputGenerator(0, T, dt)
data = i_gen.generate_lorenz()
num_train = int(len(data.T) * RATIO_TRAIN)
train_data = data.T[:num_train]
print(f"train_data_shape: {train_data.shape}")
model = ReservoirNetWork(inputs=train_data,
num_input_nodes=NUM_INPUT_NODES,
num_reservoir_nodes=NUM_RESERVOIR_NODES,
num_output_nodes=NUM_OUTPUT_NODES,
leak_rate=LEAK_RATE)
model.train()
train_result = model.get_train_result() # 訓練の結果を取得
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.scatter(data[0], data[1], data[2], marker='o')
ax.set_title("Echo State Network Sin Prediction")
ax.set_xlabel("x")
ax.set_ylabel("y")
ax.set_zlabel("z")
plt.show()