cwpearson/suitesparse-downloader
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add square datasets

Browse Source
This commit is contained in:
Carl William Pearson
2022-01-28 13:14:30 -07:00
parent 29f5289068
commit 4584ce14ee
1 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
lib/datasets.py
View File

@@ -36,6 +36,9 @@ def filter_reject_large(mats):
def filter_reject_small(mats): def filter_reject_small(mats):
return [mat for mat in mats if not mat_is_small(mat)] return [mat for mat in mats if not mat_is_small(mat)]
def filter_keep_square(mats):
return [mat for mat in mats if mat.rows == mat.cols]
## all real-valued matrices ## all real-valued matrices
REAL_MATS = Dataset( REAL_MATS = Dataset(
name = "reals", name = "reals",
@@ -80,6 +83,15 @@ REAL_SMALL_MATS = Dataset (
mats = filter_reject_large(REAL_MATS.mats) mats = filter_reject_large(REAL_MATS.mats)
) )
REGULAR_SQUARE_REAL_SMALL_MATS = Dataset (
name = "regular_square_reals_small",
mats = filter_keep_square(REGULAR_REAL_SMALL_MATS.mats)
)
SQUARE_REAL_SMALL_MATS = Dataset (
name = "square_reals_small",
mats = filter_keep_square(REAL_SMALL_MATS.mats)
)
## keep "medium" matrices ## keep "medium" matrices
REGULAR_REAL_MED_MATS = Dataset ( REGULAR_REAL_MED_MATS = Dataset (
name = "regular_reals_med", name = "regular_reals_med",
@@ -97,7 +109,9 @@ DATASETS = [
REAL_MED_MATS, REAL_MED_MATS,
# REGULAR_REAL_MATS, # REGULAR_REAL_MATS,
REGULAR_REAL_SMALL_MATS, REGULAR_REAL_SMALL_MATS,
REGULAR_REAL_MED_MATS REGULAR_REAL_MED_MATS,
REGULAR_SQUARE_REAL_SMALL_MATS,
SQUARE_REAL_SMALL_MATS,
] ]
def get_kinds(): def get_kinds():