hummingbird-deepnote_2.py

hummingbird-deepnote_2.py

import torch
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
import mpl_toolkits
from sklearn.datasets import make_classification
from sklearn.model_selection import train_test_split
from sklearn.ensemble import RandomForestRegressor
from hummingbird.ml import convert

print('GPU available: %s' % torch.cuda.is_available())

%matplotlib inline