 #666666">.</span>reset_index( drop <span style="color: #666666">=</span> <span style="color: #008000">True</span> )

<span style="color: #408080; font-style: italic"># Creating the X, T and E input</span>
X_train, X_test <span style="color: #666666">=</span> data_train[features], data_test[features]
T_train, T_test <span style="color: #666666">=</span> data_train[<span style="color: #BA2121">&#39;time&#39;</span>]<span style="color: #666666">.</span>values, data_test[<span style="color: #BA2121">&#39;time&