Skip to content

Commit

Permalink
updated tests for multidataset
Browse files Browse the repository at this point in the history
  • Loading branch information
raamana committed Apr 3, 2020
1 parent 4a5ccf3 commit 93270d4
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 12 deletions.
3 changes: 2 additions & 1 deletion pyradigm/multiple.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
import random
from warnings import warn
from collections import Iterable, Counter
from collections.abc import Iterable
from collections import Counter
from copy import copy
from operator import itemgetter
from sys import version_info
Expand Down
21 changes: 10 additions & 11 deletions pyradigm/tests/test_MultiDataset.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,14 @@
import numpy as np
from os.path import join as pjoin, exists as pexists, realpath, dirname
from os import makedirs
from functools import partial
from pyradigm import (MultiDatasetClassify, MultiDatasetRegress,
ClassificationDataset as ClfDataset,
RegressionDataset as RegrDataset)
from pyradigm.utils import (make_random_ClfDataset, make_random_RegrDataset,
make_random_dataset,
dataset_with_new_features_same_everything_else)
from os import makedirs
from os.path import dirname, join as pjoin, realpath

import numpy as np

from pyradigm import (ClassificationDataset as ClfDataset, MultiDatasetClassify,
MultiDatasetRegress, RegressionDataset as RegrDataset)
from pyradigm.utils import (dataset_with_new_features_same_everything_else,
make_random_ClfDataset, make_random_RegrDataset,
make_random_dataset)

test_dir = dirname(__file__)
out_dir = realpath(pjoin(test_dir, 'tmp'))
Expand All @@ -17,8 +18,6 @@
max_num_modalities = 10
max_feat_dim = 10

ds_class = RegrDataset


def make_fully_separable_classes(max_class_size=10, max_dim=22):
from sklearn.datasets import make_blobs
Expand Down

0 comments on commit 93270d4

Please sign in to comment.