Matheus Xavier Sampaio mxaviersmp
regispires
/ h2o_utils.py
Last active
November 9, 2021 03:21
Utility functions to work with the h2o machine learning framework
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import h2o | |
| def h2o_train(model, X_train, y_train, X_val=None, y_val=None): | |
| print('h2o_train - training...') | |
| X_y_train = np.c_[ X_train, y_train ] | |
| frame_train = h2o.H2OFrame(X_y_train) | |
| if X_val is not None and y_val is not None: | |
| X_y_val = np.c_[ X_val, y_val ] | |
| frame_val = h2o.H2OFrame(X_y_val) |