text_plain_output_1.png
x_train = x_train.reshape(len(x_train), 200 * 200)
x_test = x_test.reshape(len(x_test), 200 * 200)
x_val = x_val.reshape(len(x_val), 200 * 200)

cuml_model = cuRFC(max_features=1.0, n_bins=8, n_estimators=40)
cuml_model.fit(x_train.astype('float32'), y_train.astype('float32'))