Skip to content

Commit

Permalink
Replacing csd in sample selection with california because the larger …
Browse files Browse the repository at this point in the history
…dataset means a few redundant samples, which is not good for stable testing
  • Loading branch information
rosecers committed May 16, 2022
1 parent ba1a32c commit a42d2a3
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions tests/test_sample_simple_cur.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,14 +3,15 @@
import numpy as np
from sklearn import exceptions

from skcosmo.datasets import load_csd_1000r as load
from sklearn.datasets import fetch_california_housing as load
from skcosmo.sample_selection import CUR


class TestCUR(unittest.TestCase):
def setUp(self):
self.X, self.y = load(return_X_y=True)
self.n_select = 20
self.X, _ = load(return_X_y=True)
self.X = self.X[:1000]
self.n_select = min(20, min(self.X.shape) // 2)

def test_bad_transform(self):
selector = CUR(n_to_select=2)
Expand Down

0 comments on commit a42d2a3

Please sign in to comment.