Hi,
I would be very interested in testing this architecture on another dataset but I found the current API very focused on the task of testing on the collection you are using and difficult to generalize to others (at least for me).
Wouldn't it be possible (and very nice) if there was a class with an API compatible with sklearn with a fit(X_3dnumpyarray, y_1dnumpyarray) method that returns the fitted estimator itself and a predict method that returns the 1d predicted numpy array?
Thank you