def setUp(self):
    n_features = 3
    device = 'cpu'
    self.df = get_df()
    self.df.drop('target', axis=1, inplace=True)
    set_random_states()
def setUp(self):
    n_features = 3
    device = 'cpu'
    self.df = get_df()
    self.df.drop('target', axis=1, inplace=True)
    set_random_states()