radMLBench 1.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
radMLBench/__init__.py ADDED
@@ -0,0 +1,32 @@
1
+ # -*- coding: utf-8 -*-
2
+
3
+ """
4
+ RadMLBench was developed at the University Hospital in Essen, Germany.
5
+ For questions use github issues or write an email (aydin.demircioglu@uk-essen.de).
6
+
7
+ RadMLBench is partially based on the PMLB (Penn Machine Learning Benchmarks), see https://epistasislab.github.io/pmlb/
8
+
9
+ Permission is hereby granted, free of charge, to any person obtaining a copy of this software
10
+ and associated documentation files (the "Software"), to deal in the Software without restriction,
11
+ including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense,
12
+ and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so,
13
+ subject to the following conditions:
14
+
15
+ The above copyright notice and this permission notice shall be included in all copies or substantial
16
+ portions of the Software.
17
+
18
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT
19
+ LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
20
+ IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
21
+ WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
22
+ SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
23
+ """
24
+
25
+ from .radMLBench import (
26
+ listDatasets,
27
+ loadData,
28
+ getCVSplits,
29
+ getMetaData)
30
+
31
+
32
+ #
@@ -0,0 +1,976 @@
1
+ Ahn2021:
2
+ BrokenFeatures: 0
3
+ ClassBalance: 55
4
+ Dimensionality: 0.96
5
+ Missings: 0
6
+ authors: Choi et al.
7
+ dataset: Ahn2021
8
+ download_source: https://data.mendeley.com/datasets/9mpy8mc6hn/1
9
+ modality: MRI
10
+ nFeatures: 108
11
+ nInstances: 114
12
+ outcome: MGMT mutation status
13
+ pathology: Glioblastoma
14
+ publication_doi: https://doi.org/10.1016/j.ejrad.2019.108642
15
+ publication_title: 'Analysis of heterogeneity of peritumoral T2 hyperintensity in
16
+ patients with pretreatment glioblastoma: Prognostic value of MRI-based radiomics'
17
+ remarks: null
18
+ software: PyRadiomics
19
+ year: 2021
20
+ Arita2018:
21
+ BrokenFeatures: 0
22
+ ClassBalance: 66
23
+ Dimensionality: 4.08
24
+ Missings: 168
25
+ authors: Arita et al.
26
+ dataset: Arita2018
27
+ download_source: https://www.nature.com/articles/s41598-018-30273-4#Sec18
28
+ modality: MRI
29
+ nFeatures: 683
30
+ nInstances: 168
31
+ outcome: IDH mutation status
32
+ pathology: Glioma
33
+ publication_doi: https://doi.org/10.1038/s41598-018-30273-4
34
+ publication_title: Lesion location implemented magnetic resonance imaging radiomics
35
+ for predicting IDH and TERT promoter mutations in grade II/III gliomas
36
+ remarks: null
37
+ software: Inhouse
38
+ year: 2018
39
+ BraTS-2021:
40
+ BrokenFeatures: 0
41
+ ClassBalance: 52
42
+ Dimensionality: 7.04
43
+ Missings: 0
44
+ authors: Flanders et al.
45
+ dataset: BraTS-2021
46
+ download_source: https://radiomics.uk/
47
+ modality: MRI
48
+ nFeatures: 4060
49
+ nInstances: 577
50
+ outcome: MGMT mutation status
51
+ pathology: Glioblastoma
52
+ publication_doi: https://doi.org/10.48550/arXiv.2107.02314
53
+ publication_title: The RSNA-ASNR-MICCAI BraTS 2021 Benchmark on Brain Tumor Segmentation
54
+ and Radiogenomic Classification
55
+ remarks: Original dataset at https://www.kaggle.com/c/rsna-miccai-brain-tumor-radiogenomic-classification
56
+ software: PyRadiomics
57
+ year: 2021
58
+ Brancato2023:
59
+ BrokenFeatures: 0
60
+ ClassBalance: 60
61
+ Dimensionality: 41.07
62
+ Missings: 0
63
+ authors: Castaldo et al.
64
+ dataset: Brancato2023
65
+ download_source: https://data.mendeley.com/datasets/dvysdynchh/1
66
+ modality: MRI
67
+ nFeatures: 2380
68
+ nInstances: 58
69
+ outcome: Clinical significance
70
+ pathology: Prostate cancer
71
+ publication_doi: https://doi.org/10.3390/jcm12010140
72
+ publication_title: A Framework of Analysis to Facilitate the Harmonization of Multicenter
73
+ Radiomic Features in Prostate Cancer
74
+ remarks: Only center 2 was used, since no labels were available for center 1
75
+ software: PyRadiomics
76
+ year: 2023
77
+ C4KC-KiTS:
78
+ BrokenFeatures: 0
79
+ ClassBalance: 66
80
+ Dimensionality: 4.53
81
+ Missings: 0
82
+ authors: Heller et al.
83
+ dataset: C4KC-KiTS
84
+ download_source: https://radiomics.uk/
85
+ modality: CT
86
+ nFeatures: 315
87
+ nInstances: 70
88
+ outcome: Histological subtype
89
+ pathology: Kidney tumor
90
+ publication_doi: https://doi.org/10.1016/j.media.2020.101821
91
+ publication_title: 'The state of the art in kidney and kidney tumor segmentation
92
+ in contrast-enhanced CT imaging: Results of the KiTS19 challenge'
93
+ remarks: Original dataset at https://www.cancerimagingarchive.net/collection/c4kc-kits.
94
+ software: PyRadiomics
95
+ year: 2021
96
+ Colorectal-Liver-Metastases:
97
+ BrokenFeatures: 0
98
+ ClassBalance: 84
99
+ Dimensionality: 5.86
100
+ Missings: 0
101
+ authors: Simpson et al.
102
+ dataset: Colorectal-Liver-Metastases
103
+ download_source: https://radiomics.uk/
104
+ modality: CT
105
+ nFeatures: 525
106
+ nInstances: 90
107
+ outcome: Survival (at 10 years)
108
+ pathology: Colorectal cancer
109
+ publication_doi: https://doi.org/10.1038/s41597-024-02981-2
110
+ publication_title: Preoperative CT and survival data for patients undergoing resection
111
+ of colorectal liver metastases
112
+ remarks: Original dataset at https://www.cancerimagingarchive.net/collection/colorectal-liver-metastases.
113
+ Only patient that died were included.
114
+ software: PyRadiomics
115
+ year: 2024
116
+ Dai2023:
117
+ BrokenFeatures: 0
118
+ ClassBalance: 26
119
+ Dimensionality: 7.17
120
+ Missings: 0
121
+ authors: Dai et al.
122
+ dataset: Dai2023
123
+ download_source: https://peerj.com/articles/16230/#supplemental-information
124
+ modality: CT
125
+ nFeatures: 851
126
+ nInstances: 119
127
+ outcome: Thrombocytopenia presence
128
+ pathology: Gastrointestinal malignancies
129
+ publication_doi: https://doi.org/10.7717/peerj.16230
130
+ publication_title: 'A contrast-enhanced CT-based whole-spleen radiomics signature
131
+ for early prediction of oxaliplatin-related thrombocytopenia in patients with
132
+ gastrointestinal malignancies: a retrospective study'
133
+ remarks: null
134
+ software: PyRadiomics
135
+ year: 2023
136
+ Deng2023:
137
+ BrokenFeatures: 1
138
+ ClassBalance: 36
139
+ Dimensionality: 0.87
140
+ Missings: 0
141
+ authors: Deng et al.
142
+ dataset: Deng2023
143
+ download_source: https://github.com/BboyT/BM_NSCLC_subpathology
144
+ modality: MRI
145
+ nFeatures: 225
146
+ nInstances: 261
147
+ outcome: Pathological diagnosis
148
+ pathology: NSCLC
149
+ publication_doi: https://doi.org/10.1007/s13246-023-01300-0
150
+ publication_title: 'MRI radiomics for brain metastasis sub-pathology classification
151
+ from non-small cell lung cancer: a machine learning, multicenter study'
152
+ remarks: All data were merged. Attached python applies CV 'wrongly' on internal
153
+ data, but there is external data, so test results are unbiased.
154
+ software: PyRadiomics
155
+ year: 2023
156
+ Dong2022:
157
+ BrokenFeatures: 0
158
+ ClassBalance: 49
159
+ Dimensionality: 3.06
160
+ Missings: 0
161
+ authors: Dong et al.
162
+ dataset: Dong2022
163
+ download_source: https://peerj.com/articles/14127/#supplemental-information
164
+ modality: CT
165
+ nFeatures: 851
166
+ nInstances: 279
167
+ outcome: Tuberculosis granuloma presence
168
+ pathology: Lung cancer
169
+ publication_doi: https://doi.org/10.7717/peerj.14127
170
+ publication_title: Radiomics combined with clinical features in distinguishing non-calcifying
171
+ tuberculosis granuloma and lung adenocarcinoma in small pulmonary nodules
172
+ remarks: null
173
+ software: PyRadiomics (FAE)
174
+ year: 2022
175
+ Fusco2022:
176
+ BrokenFeatures: 0
177
+ ClassBalance: 61
178
+ Dimensionality: 3.59
179
+ Missings: 0
180
+ authors: Fusco et al.
181
+ dataset: Fusco2022
182
+ download_source: https://zenodo.org/records/6344730
183
+ modality: MRI
184
+ nFeatures: 192
185
+ nInstances: 54
186
+ outcome: Malignancy presence
187
+ pathology: Breast cancer
188
+ publication_doi: https://doi.org/10.3390/curroncol29030159
189
+ publication_title: Radiomic and Artificial Intelligence Analysis with Textural Metrics
190
+ Extracted by Contrast-Enhanced Mammography and Dynamic Contrast Magnetic Resonance
191
+ Imaging to Detect Breast Malignant Lesions
192
+ remarks: null
193
+ software: TextureToolbox
194
+ year: 2022
195
+ Granata2021:
196
+ BrokenFeatures: 3
197
+ ClassBalance: 47
198
+ Dimensionality: 6.61
199
+ Missings: 0
200
+ authors: Granata et al.
201
+ dataset: Granata2021
202
+ download_source: https://zenodo.org/records/5162861
203
+ modality: CT
204
+ nFeatures: 580
205
+ nInstances: 88
206
+ outcome: Overall survival (at 32 months)
207
+ pathology: Lung adenocarcinoma
208
+ publication_doi: https://doi.org/10.3390/cancers13163992
209
+ publication_title: Preliminary Report on Computed Tomography Radiomics Features
210
+ as Biomarkers to Immunotherapy Selection in Lung Adenocarcinoma Patients
211
+ remarks: No dead status/event is given.
212
+ software: HealthMyne (Commercial)
213
+ year: 2021
214
+ Granata2024:
215
+ BrokenFeatures: 0
216
+ ClassBalance: 75
217
+ Dimensionality: 16.73
218
+ Missings: 0
219
+ authors: Granata et al.
220
+ dataset: Granata2024
221
+ download_source: https://zenodo.org/records/10464602
222
+ modality: MRI
223
+ nFeatures: 851
224
+ nInstances: 51
225
+ outcome: Tumor budding presence
226
+ pathology: Colorectal cancer
227
+ publication_doi: https://doi.org/10.3390/diagnostics14020152
228
+ publication_title: Machine Learning and Radiomics Analysis for Tumor Budding Prediction
229
+ in Colorectal Liver Metastases Magnetic Resonance Imaging Assessment
230
+ remarks: null
231
+ software: PyRadiomics
232
+ year: 2024
233
+ HCC-TACE-Seg:
234
+ BrokenFeatures: 0
235
+ ClassBalance: 14
236
+ Dimensionality: 5.02
237
+ Missings: 0
238
+ authors: Moawad et al.
239
+ dataset: HCC-TACE
240
+ download_source: https://radiomics.uk/
241
+ modality: CT
242
+ nFeatures: 420
243
+ nInstances: 84
244
+ outcome: Lymph node metastasis presence
245
+ pathology: Hepatocellular cancer
246
+ publication_doi: https://doi.org/10.1038/s41597-023-01928-3
247
+ publication_title: Multimodality annotated hepatocellular carcinoma data set including
248
+ pre- and post-TACE with imaging segmentation
249
+ remarks: Original data at https://www.cancerimagingarchive.net/collection/hcc-tace-seg/.
250
+ software: PyRadiomics
251
+ year: 2023
252
+ HNSCC:
253
+ BrokenFeatures: 0
254
+ ClassBalance: 27
255
+ Dimensionality: 1.15
256
+ Missings: 0
257
+ authors: Grossberg et al.
258
+ dataset: HNSCC
259
+ download_source: https://radiomics.uk/
260
+ modality: CT
261
+ nFeatures: 105
262
+ nInstances: 93
263
+ outcome: Lymph node metastasis presence
264
+ pathology: Head-and-neck cancer
265
+ publication_doi: https://doi.org/10.1038/sdata.2018.173
266
+ publication_title: Imaging and Clinical Data Archive for Head and Neck Squamous
267
+ Cell Carcinoma Patients Treated with Radiotherapy
268
+ remarks: Original data at https://www.cancerimagingarchive.net/collection/hnscc/.
269
+ software: PyRadiomics
270
+ year: 2018
271
+ Head-Neck-PET-CT:
272
+ BrokenFeatures: 0
273
+ ClassBalance: 67
274
+ Dimensionality: 2.33
275
+ Missings: 0
276
+ authors: "Valli\xE8res et al."
277
+ dataset: Head-Neck-PET-CT
278
+ download_source: https://radiomics.uk/
279
+ modality: PET/CT
280
+ nFeatures: 210
281
+ nInstances: 91
282
+ outcome: Lymph node metastasis presence
283
+ pathology: Head-and-neck cancer
284
+ publication_doi: https://doi.org/10.1038/s41598-017-10371-5
285
+ publication_title: Radiomics strategies for risk assessment of tumour failure in
286
+ head-and-neck cancer
287
+ remarks: Original data at https://www.cancerimagingarchive.net/collection/head-neck-pet-ct/.
288
+ software: PyRadiomics
289
+ year: 2017
290
+ Head-Neck-Radiomics-HN1:
291
+ BrokenFeatures: 0
292
+ ClassBalance: 45
293
+ Dimensionality: 0.78
294
+ Missings: 0
295
+ authors: Aerts et al.
296
+ dataset: Head-Neck-Radiomics-HN1
297
+ download_source: https://radiomics.uk/
298
+ modality: CT
299
+ nFeatures: 105
300
+ nInstances: 137
301
+ outcome: Lymph node metastasis presence
302
+ pathology: Head-and-neck cancer
303
+ publication_doi: http://doi.org/10.1038/ncomms5006
304
+ publication_title: Decoding Tumour Phenotype by Noninvasive Imaging Using a Quantitative
305
+ Radiomics Approach
306
+ remarks: Original data at https://www.cancerimagingarchive.net/collection/head-neck-radiomics-hn1/
307
+ software: PyRadiomics
308
+ year: 2014
309
+ Hosny2018A:
310
+ BrokenFeatures: 1
311
+ ClassBalance: 54
312
+ Dimensionality: 3.37
313
+ Missings: 2258
314
+ authors: Hosny et al.
315
+ dataset: Hosny2018A
316
+ download_source: https://github.com/modelhub-ai/deep-prognosis
317
+ modality: CT
318
+ nFeatures: 984
319
+ nInstances: 293
320
+ outcome: Overall survival (at 2 years)
321
+ pathology: NSCLC
322
+ publication_doi: https://doi.org/10.1371/journal.pmed.1002711
323
+ publication_title: 'Deep learning for lung cancer prognostication: A retrospective
324
+ multi-cohort radiomics study'
325
+ remarks: HarvardRT subset. Removed exponential_glcm features since they exhibited
326
+ NA values due to numerical overflow.
327
+ software: PyRadiomics
328
+ year: 2018
329
+ Hosny2018B:
330
+ BrokenFeatures: 1
331
+ ClassBalance: 29
332
+ Dimensionality: 4.76
333
+ Missings: 1157
334
+ authors: Hosny et al.
335
+ dataset: Hosny2018B
336
+ download_source: https://github.com/modelhub-ai/deep-prognosis
337
+ modality: CT
338
+ nFeatures: 984
339
+ nInstances: 207
340
+ outcome: Overall survival (at 2 years)
341
+ pathology: NSCLC
342
+ publication_doi: https://doi.org/10.1371/journal.pmed.1002711
343
+ publication_title: 'Deep learning for lung cancer prognostication: A retrospective
344
+ multi-cohort radiomics study'
345
+ remarks: Maastro subset. Removed exponential_glcm features since they exhibited
346
+ NA values due to numerical overflow.
347
+ software: PyRadiomics
348
+ year: 2018
349
+ Hosny2018C:
350
+ BrokenFeatures: 1
351
+ ClassBalance: 73
352
+ Dimensionality: 5.39
353
+ Missings: 169
354
+ authors: Hosny et al.
355
+ dataset: Hosny2018C
356
+ download_source: https://github.com/modelhub-ai/deep-prognosis
357
+ modality: CT
358
+ nFeatures: 984
359
+ nInstances: 183
360
+ outcome: Overall survival (at 2 years)
361
+ pathology: NSCLC
362
+ publication_doi: https://doi.org/10.1371/journal.pmed.1002711
363
+ publication_title: 'Deep learning for lung cancer prognostication: A retrospective
364
+ multi-cohort radiomics study'
365
+ remarks: Moffitt subset. Removed exponential_glcm features since they exhibited
366
+ NA values due to numerical overflow.
367
+ software: PyRadiomics
368
+ year: 2018
369
+ Huang2023:
370
+ BrokenFeatures: 0
371
+ ClassBalance: 46
372
+ Dimensionality: 4.04
373
+ Missings: 0
374
+ authors: Huang et al.
375
+ dataset: Huang2023
376
+ download_source: https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0292110#sec019
377
+ modality: CT
378
+ nFeatures: 855
379
+ nInstances: 212
380
+ outcome: Malignancy presence
381
+ pathology: Renal cancer
382
+ publication_doi: https://doi.org/10.1371/journal.pone.0292110
383
+ publication_title: 'Enhanced and unenhanced: Radiomics models for discriminating
384
+ between benign and malignant cystic renal masses on CT images: A multi-center
385
+ study'
386
+ remarks: Unenhanced CT features were used
387
+ software: PyRadiomics
388
+ year: 2023
389
+ Hunter2023:
390
+ BrokenFeatures: 0
391
+ ClassBalance: 54
392
+ Dimensionality: 3.85
393
+ Missings: 0
394
+ authors: Hunter et al.
395
+ dataset: Hunter2023
396
+ download_source: https://data.mendeley.com/datasets/rxn95mp24d/1
397
+ modality: CT
398
+ nFeatures: 1998
399
+ nInstances: 520
400
+ outcome: Nodule malignancy presence
401
+ pathology: Lung cancer
402
+ publication_doi: https://doi.org/10.1038/s41416-023-02480-y
403
+ publication_title: Radiomics-based decision support tool assists radiologists in
404
+ small lung nodule classification and improves lung cancer early diagnosis
405
+ remarks: null
406
+ software: TexLab
407
+ year: 2023
408
+ ISPY1:
409
+ BrokenFeatures: 9
410
+ ClassBalance: 57
411
+ Dimensionality: 2.31
412
+ Missings: 0
413
+ authors: Newitt et al.
414
+ dataset: ISPY1
415
+ download_source: https://www.cancerimagingarchive.net/analysis-result/ispy1-tumor-seg-radiomics/
416
+ modality: MRI
417
+ nFeatures: 370
418
+ nInstances: 161
419
+ outcome: Hormone receptor positive status
420
+ pathology: Breast cancer
421
+ publication_doi: http://doi.org/10.7937/K9/TCIA.2016.HdHpgJLK
422
+ publication_title: Multi-center breast DCE-MRI data and segmentations from patients
423
+ in the I-SPY 1/ACRIN 6657 trials
424
+ remarks: TCIA sources at https://www.cancerimagingarchive.net/collection/ispy1/,
425
+ features from the cited analysis were used.
426
+ software: CaPTk
427
+ year: 2016
428
+ Keek2020:
429
+ BrokenFeatures: 10
430
+ ClassBalance: 44
431
+ Dimensionality: 4.85
432
+ Missings: 52
433
+ authors: Keek et al.
434
+ dataset: Keek2020
435
+ download_source: https://github.com/SebastianSanduleanu/Peritumoral-HN-Radiomics
436
+ modality: CT
437
+ nFeatures: 1322
438
+ nInstances: 273
439
+ outcome: Overall survival (at 3 years)
440
+ pathology: HNSCC
441
+ publication_doi: https://doi.org/10.1371/journal.pone.0232639
442
+ publication_title: Computed tomography-derived radiomic signature of head and neck
443
+ squamous cell carcinoma (peri)tumoral tissue for the prediction of locoregional
444
+ recurrence and distant metastasis after concurrent chemo-radiotherapy
445
+ remarks: Due to incompatibilities, only the DESIGN data was used. Patients with
446
+ less than 3 years of follow-up were removed.
447
+ software: Inhouse
448
+ year: 2020
449
+ LGG-1p19qDeletion:
450
+ BrokenFeatures: 0
451
+ ClassBalance: 64
452
+ Dimensionality: 12.78
453
+ Missings: 0
454
+ authors: Akkus et al.
455
+ dataset: LGG-1p19qDeletion
456
+ download_source: https://radiomics.uk/
457
+ modality: MRI
458
+ nFeatures: 2030
459
+ nInstances: 159
460
+ outcome: 1p19q co-deletion status
461
+ pathology: Glioma
462
+ publication_doi: https://doi.org/10.1007/s10278-017-9984-3
463
+ publication_title: Predicting Deletion of Chromosomal Arms 1p/19q in Low-Grade Gliomas
464
+ from MR Images Using Machine Intelligence
465
+ remarks: Original data at https://www.cancerimagingarchive.net/collection/lgg-1p19qdeletion/
466
+ software: PyRadiomics
467
+ year: 2017
468
+ LNDb:
469
+ BrokenFeatures: 0
470
+ ClassBalance: 66
471
+ Dimensionality: 0.62
472
+ Missings: 1
473
+ authors: Pedrosa et al.
474
+ dataset: LNDb
475
+ download_source: https://radiomics.uk/
476
+ modality: CT
477
+ nFeatures: 105
478
+ nInstances: 173
479
+ outcome: Fleischner score group
480
+ pathology: Lung cancer
481
+ publication_doi: https://doi.org/10.48550/arXiv.1911.08434
482
+ publication_title: 'LNDb: a lung nodule database on computed tomography'
483
+ remarks: Original dataset et https://zenodo.org/records/6613714#.Y8vLZezMIUq
484
+ software: PyRadiomics
485
+ year: 2019
486
+ Li2020:
487
+ BrokenFeatures: 0
488
+ ClassBalance: 63
489
+ Dimensionality: 7.8
490
+ Missings: 0
491
+ authors: Li et al.
492
+ dataset: Li2020
493
+ download_source: https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0227703#sec017
494
+ modality: MRI
495
+ nFeatures: 396
496
+ nInstances: 51
497
+ outcome: Immunohistochemical result
498
+ pathology: Glioma
499
+ publication_doi: https://doi.org/10.1371/journal.pone.0227703
500
+ publication_title: 'High-order radiomics features based on T2 FLAIR MRI predict
501
+ multiple glioma immunohistochemical features: A more precise and personalized
502
+ gliomas management'
503
+ remarks: Outcome is based on Ki-67, S-100, vimentin and CD34 immunohistochemical
504
+ results (called Label in the study)
505
+ software: Artificial Intelligence Kit (Commercial)
506
+ year: 2020
507
+ Lu2019:
508
+ BrokenFeatures: 0
509
+ ClassBalance: 73
510
+ Dimensionality: 8.79
511
+ Missings: 0
512
+ authors: Lu et al.
513
+ dataset: Lu2019
514
+ download_source: https://doi.org/10.17632/4c5znk5m8t.1
515
+ modality: CT
516
+ nFeatures: 657
517
+ nInstances: 75
518
+ outcome: Progression free survival (at 2 years)
519
+ pathology: Ovarian cancer
520
+ publication_doi: https://doi.org/10.1038/s41467-019-08718-9
521
+ publication_title: A mathematical-descriptor of tumor-mesoscopic-structure from
522
+ computed-tomography images annotates prognostic- and molecular-phenotypes of epithelial
523
+ ovarian cancer
524
+ remarks: Only patients with follow-up at least 2 years were included, only those
525
+ with bilateral tumors (Code > 1).
526
+ software: Inhouse
527
+ year: 2019
528
+ Meningioma-SEG-CLASS:
529
+ BrokenFeatures: 0
530
+ ClassBalance: 43
531
+ Dimensionality: 23.09
532
+ Missings: 0
533
+ authors: Vassantachart et al.
534
+ dataset: Meningioma-SEG-CLASS
535
+ download_source: https://radiomics.uk/
536
+ modality: MRI
537
+ nFeatures: 2030
538
+ nInstances: 88
539
+ outcome: Tumor grade
540
+ pathology: Meningioma
541
+ publication_doi: https://doi.org/10.1038/s41598-022-07859-0
542
+ publication_title: Automatic differentiation of Grade I and II meningiomas on magnetic
543
+ resonance image using an asymmetric convolutional neural network
544
+ remarks: Original dataset at https://www.cancerimagingarchive.net/collection/meningioma-seg-class/
545
+ software: PyRadiomics
546
+ year: 2022
547
+ NSCLC-Radiogenomics:
548
+ BrokenFeatures: 0
549
+ ClassBalance: 16
550
+ Dimensionality: 0.74
551
+ Missings: 0
552
+ authors: Gevaert et al.
553
+ dataset: NSCLC-Radiogenomics
554
+ download_source: https://radiomics.uk/
555
+ modality: PET/CT
556
+ nFeatures: 105
557
+ nInstances: 144
558
+ outcome: EGFR mutation status
559
+ pathology: NSCLC
560
+ publication_doi: http://doi.org/10.1148/radiol.12111607
561
+ publication_title: "Non\u2013Small Cell Lung Cancer: Identifying Prognostic Imaging\
562
+ \ Biomarkers by Leveraging Public Gene Expression Microarray Data\u2014Methods\
563
+ \ and Preliminary Results"
564
+ remarks: Original dataset at https://www.cancerimagingarchive.net/collection/nsclc-radiogenomics/
565
+ software: PyRadiomics
566
+ year: 2012
567
+ OcanaTienda2023:
568
+ BrokenFeatures: 0
569
+ ClassBalance: 48
570
+ Dimensionality: 16.9
571
+ Missings: 0
572
+ authors: "Oca\xF1a-Tienda et al."
573
+ dataset: OcanaTienda2023
574
+ download_source: https://springernature.figshare.com/articles/dataset/Radiomic_data_of_annotated_brain_metastases/22277638/1
575
+ modality: MRI
576
+ nFeatures: 1130
577
+ nInstances: 67
578
+ outcome: Survival (at 2 years)
579
+ pathology: Brain metastasis
580
+ publication_doi: https://doi.org/10.1038/s41597-023-02123-0
581
+ publication_title: A comprehensive dataset of annotated brain metastasis MR images
582
+ with clinical and radiomic data
583
+ remarks: All patients died, therefore all data was processed.
584
+ software: PyRadiomics
585
+ year: 2023
586
+ PI-CAI:
587
+ BrokenFeatures: 0
588
+ ClassBalance: 66
589
+ Dimensionality: 3.14
590
+ Missings: 0
591
+ authors: Saha et al.
592
+ dataset: PI-CAI
593
+ download_source: https://radiomics.uk/
594
+ modality: MRI
595
+ nFeatures: 3045
596
+ nInstances: 969
597
+ outcome: Gleason score risk group
598
+ pathology: Prostate cancer
599
+ publication_doi: https://doi.org/10.1016/j.media.2021.102155
600
+ publication_title: 'End-to-end prostate cancer detection in bpMRI via 3D CNNs: Effects
601
+ of attention mechanisms, clinical priori and decoupled false positive reduction'
602
+ remarks: Original dataset at https://pi-cai.grand-challenge.org/
603
+ software: PyRadiomics
604
+ year: 2021
605
+ Petrillo2023:
606
+ BrokenFeatures: 0
607
+ ClassBalance: 37
608
+ Dimensionality: 6.66
609
+ Missings: 0
610
+ authors: Petrillo et al.
611
+ dataset: Petrillo2023
612
+ download_source: https://zenodo.org/records/8392800
613
+ modality: MRI
614
+ nFeatures: 851
615
+ nInstances: 128
616
+ outcome: Luminal type presence
617
+ pathology: Breast cancer
618
+ publication_doi: https://doi.org/10.1007/s11547-023-01718-2
619
+ publication_title: Radiomics and artificial intelligence analysis by T2-weighted
620
+ imaging and dynamic contrast-enhanced magnetic resonance imaging to predict Breast
621
+ Cancer Histological Outcome
622
+ remarks: Since luminal vs non-luminal had highest AUC, this was selected as endpoint.
623
+ All features (T2, DCE) were merged.
624
+ software: PyRadimomics
625
+ year: 2023
626
+ Prostate-MRI-US-Biopsy:
627
+ BrokenFeatures: 0
628
+ ClassBalance: 77
629
+ Dimensionality: 1.32
630
+ Missings: 0
631
+ authors: Sonn et al.
632
+ dataset: Prostate-MRI-US-Biopsy
633
+ download_source: https://radiomics.uk/
634
+ modality: MRI
635
+ nFeatures: 1015
636
+ nInstances: 773
637
+ outcome: Gleason score risk group
638
+ pathology: Prostate cancer
639
+ publication_doi: https://doi.org/10.1016/j.juro.2012.08.095
640
+ publication_title: Targeted biopsy in the detection of prostate cancer using an
641
+ office based magnetic resonance ultrasound fusion device
642
+ remarks: Original dataset at https://www.cancerimagingarchive.net/collection/prostate-mri-us-biopsy/
643
+ software: PyRadiomics
644
+ year: 2013
645
+ QIN-HEADNECK:
646
+ BrokenFeatures: 0
647
+ ClassBalance: 75
648
+ Dimensionality: 3.59
649
+ Missings: 0
650
+ authors: Fedorov et al.
651
+ dataset: QIN-HEADNECK
652
+ download_source: https://radiomics.uk/
653
+ modality: PET/CT
654
+ nFeatures: 210
655
+ nInstances: 59
656
+ outcome: Lymph node metastasis presence
657
+ pathology: Head-and-neck cancer
658
+ publication_doi: https://doi.org/10.7717/peerj.2057
659
+ publication_title: 'DICOM for quantitative imaging biomarker development: a standards
660
+ based approach to sharing clinical data and structured PET/CT analysis results
661
+ in head and neck cancer research.'
662
+ remarks: Original dataset at https://www.cancerimagingarchive.net/collection/qin-headneck/.
663
+ CT features were used.
664
+ software: PyRadiomics
665
+ year: 2016
666
+ Ramella2018:
667
+ BrokenFeatures: 2
668
+ ClassBalance: 55
669
+ Dimensionality: 2.68
670
+ Missings: 0
671
+ authors: Ramella et al.
672
+ dataset: Ramella2018
673
+ download_source: https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0207455#sec010
674
+ modality: CT
675
+ nFeatures: 242
676
+ nInstances: 91
677
+ outcome: Adaptive chemoradiation presence
678
+ pathology: NSCLC
679
+ publication_doi: https://doi.org/10.1371/journal.pone.0207455
680
+ publication_title: A radiomic approach for adaptive radiotherapy in non-small cell
681
+ lung cancer patients
682
+ remarks: null
683
+ software: Inhouse
684
+ year: 2018
685
+ Sasaki2019:
686
+ BrokenFeatures: 0
687
+ ClassBalance: 49
688
+ Dimensionality: 4.27
689
+ Missings: 0
690
+ authors: Sasaki et al.
691
+ dataset: Sasaki2019
692
+ download_source: https://www.nature.com/articles/s41598-019-50849-y#Sec11
693
+ modality: MRI
694
+ nFeatures: 587
695
+ nInstances: 138
696
+ outcome: MGMT mutation status
697
+ pathology: Glioblastoma
698
+ publication_doi: https://doi.org/10.1038/s41598-019-50849-y
699
+ publication_title: Radiomics and MGMT promoter methylation for prognostication of
700
+ newly diagnosed glioblastoma
701
+ remarks: null
702
+ software: Inhouse
703
+ year: 2019
704
+ Song2020:
705
+ BrokenFeatures: 0
706
+ ClassBalance: 49
707
+ Dimensionality: 1.02
708
+ Missings: 0
709
+ authors: Song et al.
710
+ dataset: Song2020
711
+ download_source: https://github.com/salan668/FAE/tree/master/Feature
712
+ modality: MRI
713
+ nFeatures: 264
714
+ nInstances: 260
715
+ outcome: Clinical significance
716
+ pathology: Prostate cancer
717
+ publication_doi: https://doi.org/10.1371/journal.pone.0237587
718
+ publication_title: 'FeAture Explorer (FAE): A tool for developing and comparing
719
+ radiomics models'
720
+ remarks: null
721
+ software: PyRadiomics
722
+ year: 2020
723
+ UCSF-PDGM:
724
+ BrokenFeatures: 0
725
+ ClassBalance: 89
726
+ Dimensionality: 17.0
727
+ Missings: 77
728
+ authors: Calabrese et al.
729
+ dataset: UCSF-PDGM
730
+ download_source: https://radiomics.uk/
731
+ modality: MRI
732
+ nFeatures: 7105
733
+ nInstances: 418
734
+ outcome: IDH mutation status
735
+ pathology: Brain cancer
736
+ publication_doi: https://doi.org/10.1148/ryai.220058
737
+ publication_title: The UCSF Preoperative Diffuse Glioma MRI (UCSF-PDGM) Dataset
738
+ remarks: Original dataset at https://www.cancerimagingarchive.net/collection/ucsf-pdgm/.
739
+ Features of the enhancing tumor were useed.
740
+ software: PyRadiomics
741
+ year: 2022
742
+ UPENN-GBM:
743
+ BrokenFeatures: 0
744
+ ClassBalance: 42
745
+ Dimensionality: 59.72
746
+ Missings: 0
747
+ authors: Bakas et al.
748
+ dataset: UPENN-GBM
749
+ download_source: https://radiomics.uk/
750
+ modality: MRI
751
+ nFeatures: 11165
752
+ nInstances: 187
753
+ outcome: MGMT mutation status
754
+ pathology: Glioblastoma
755
+ publication_doi: https://doi.org/10.1038/s41597-022-01560-7
756
+ publication_title: 'The University of Pennsylvania glioblastoma (UPenn-GBM) cohort:
757
+ advanced MRI, clinical, genomics, & radiomics'
758
+ remarks: Original data at https://www.cancerimagingarchive.net/collection/upenn-gbm/. Features
759
+ of the enhancing tumor were useed.
760
+ software: PyRadiomics
761
+ year: 2022
762
+ Veeraraghavan2020:
763
+ BrokenFeatures: 2
764
+ ClassBalance: 31
765
+ Dimensionality: 1.35
766
+ Missings: 0
767
+ authors: Veeraraghavan et al.
768
+ dataset: Veeraraghavan2020
769
+ download_source: https://github.com/The-Veeraraghavan-Lab/SciRepEndometrial2020
770
+ modality: MRI
771
+ nFeatures: 200
772
+ nInstances: 150
773
+ outcome: High tumor burden presence
774
+ pathology: Breast cancer
775
+ publication_doi: https://doi.org/10.1038/s41598-020-72475-9
776
+ publication_title: Machine learning-based prediction of microsatellite instability
777
+ and high tumor mutation burden from contrast-enhanced computed tomography in endometrial
778
+ cancers
779
+ remarks: null
780
+ software: CERR
781
+ year: 2020
782
+ WORC-CRLM:
783
+ BrokenFeatures: 0
784
+ ClassBalance: 48
785
+ Dimensionality: 13.21
786
+ Missings: 0
787
+ authors: Starmans et. al.
788
+ dataset: WORC-CRLM
789
+ download_source: https://radiomics.uk/
790
+ modality: CT
791
+ nFeatures: 1015
792
+ nInstances: 77
793
+ outcome: Histopathological growth pattern
794
+ pathology: Colorectal cancer
795
+ publication_doi: https://doi.org/10.48550/arXiv.2108.08618
796
+ publication_title: Reproducible radiomics through automated machine learning validated
797
+ on twelve clinical applications
798
+ remarks: Original data at https://xnat.bmia.nl/data/projects/worc
799
+ software: PyRadiomics
800
+ year: 2021
801
+ WORC-Desmoid:
802
+ BrokenFeatures: 0
803
+ ClassBalance: 35
804
+ Dimensionality: 5.01
805
+ Missings: 0
806
+ authors: Starmans et. al.
807
+ dataset: WORC-Desmoid
808
+ download_source: https://radiomics.uk/
809
+ modality: MRI
810
+ nFeatures: 1015
811
+ nInstances: 203
812
+ outcome: Fibromatosis presence
813
+ pathology: Soft-tissue sarcoma
814
+ publication_doi: https://doi.org/10.48550/arXiv.2108.08618
815
+ publication_title: Reproducible radiomics through automated machine learning validated
816
+ on twelve clinical applications
817
+ remarks: Original data at https://xnat.bmia.nl/data/projects/worc
818
+ software: PyRadiomics
819
+ year: 2021
820
+ WORC-GIST:
821
+ BrokenFeatures: 0
822
+ ClassBalance: 51
823
+ Dimensionality: 4.15
824
+ Missings: 0
825
+ authors: Starmans et. al.
826
+ dataset: WORC-GIST
827
+ download_source: https://radiomics.uk/
828
+ modality: CT
829
+ nFeatures: 1015
830
+ nInstances: 245
831
+ outcome: Gastrointestinal stromal tumors presence
832
+ pathology: Gastrointestinal cancer
833
+ publication_doi: https://doi.org/10.48550/arXiv.2108.08618
834
+ publication_title: Reproducible radiomics through automated machine learning validated
835
+ on twelve clinical applications
836
+ remarks: Original data at https://xnat.bmia.nl/data/projects/worc
837
+ software: PyRadiomics
838
+ year: 2021
839
+ WORC-Lipo:
840
+ BrokenFeatures: 0
841
+ ClassBalance: 50
842
+ Dimensionality: 8.92
843
+ Missings: 0
844
+ authors: Starmans et. al.
845
+ dataset: WORC-Lipo
846
+ download_source: https://radiomics.uk/
847
+ modality: MRI
848
+ nFeatures: 1015
849
+ nInstances: 114
850
+ outcome: Malignancy presence
851
+ pathology: Liposarcoma
852
+ publication_doi: https://doi.org/10.48550/arXiv.2108.08618
853
+ publication_title: Reproducible radiomics through automated machine learning validated
854
+ on twelve clinical applications
855
+ remarks: Original data at https://xnat.bmia.nl/data/projects/worc
856
+ software: PyRadiomics
857
+ year: 2021
858
+ WORC-Liver:
859
+ BrokenFeatures: 0
860
+ ClassBalance: 51
861
+ Dimensionality: 5.47
862
+ Missings: 0
863
+ authors: Starmans et. al.
864
+ dataset: WORC-Liver
865
+ download_source: https://radiomics.uk/
866
+ modality: MRI
867
+ nFeatures: 1015
868
+ nInstances: 186
869
+ outcome: Malignancy presence
870
+ pathology: Liver cancer
871
+ publication_doi: https://doi.org/10.48550/arXiv.2108.08618
872
+ publication_title: Reproducible radiomics through automated machine learning validated
873
+ on twelve clinical applications
874
+ remarks: Original data at https://xnat.bmia.nl/data/projects/worc
875
+ software: PyRadiomics
876
+ year: 2021
877
+ WORC-Melanoma:
878
+ BrokenFeatures: 0
879
+ ClassBalance: 49
880
+ Dimensionality: 10.71
881
+ Missings: 0
882
+ authors: Starmans et. al.
883
+ dataset: WORC-Melanoma
884
+ download_source: https://radiomics.uk/
885
+ modality: CT
886
+ nFeatures: 1015
887
+ nInstances: 95
888
+ outcome: BRAF mutation status
889
+ pathology: Melanoma
890
+ publication_doi: https://doi.org/10.48550/arXiv.2108.08618
891
+ publication_title: Reproducible radiomics through automated machine learning validated
892
+ on twelve clinical applications
893
+ remarks: Original data at https://xnat.bmia.nl/data/projects/worc
894
+ software: PyRadiomics
895
+ year: 2021
896
+ Wang2024:
897
+ BrokenFeatures: 0
898
+ ClassBalance: 40
899
+ Dimensionality: 4.21
900
+ Missings: 336
901
+ authors: Wang et al.
902
+ dataset: Wang2024
903
+ download_source: https://doi.org/10.6084/m9.figshare.25203245
904
+ modality: MRI
905
+ nFeatures: 280
906
+ nInstances: 67
907
+ outcome: EGFR mutation status
908
+ pathology: Glioblastoma
909
+ publication_doi: https://doi.org/10.1371/journal.pone.0299267
910
+ publication_title: Quantifying intra-tumoral genetic heterogeneity of glioblastoma
911
+ toward precision medicine using MRI and a data-inclusive machine learning algorithm
912
+ remarks: Tumoral ROIs were used. Sliding features were averaged. Multiple biopsies
913
+ were ignored by taking the first. Three samples were removed since the T2 was
914
+ missing in these.
915
+ software: Inhouse
916
+ year: 2024
917
+ Zhang2023:
918
+ BrokenFeatures: 0
919
+ ClassBalance: 51
920
+ Dimensionality: 8.78
921
+ Missings: 0
922
+ authors: Zhang et al.
923
+ dataset: Zhang2023
924
+ download_source: https://peerj.com/articles/14559/#supplemental-information
925
+ modality: CT
926
+ nFeatures: 1781
927
+ nInstances: 203
928
+ outcome: Histological invasiveness
929
+ pathology: Lung cancer
930
+ publication_doi: https://doi.org/10.7717/peerj.14559
931
+ publication_title: 'Computed tomography-based radiomics machine learning models
932
+ for prediction of histological invasiveness with sub-centimeter subsolid pulmonary
933
+ nodules: a retrospective study'
934
+ remarks: Lesions, not patient-wise data.
935
+ software: PyRadiomics
936
+ year: 2023
937
+ Zhang2024A:
938
+ BrokenFeatures: 0
939
+ ClassBalance: 57
940
+ Dimensionality: 15.11
941
+ Missings: 0
942
+ authors: Zhang et al.
943
+ dataset: Zhang2024A
944
+ download_source: https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0300170#sec017
945
+ modality: PET/CT
946
+ nFeatures: 3850
947
+ nInstances: 255
948
+ outcome: Histological type
949
+ pathology: Lung cancer
950
+ publication_doi: https://doi.org/10.1371/journal.pone.0300170
951
+ publication_title: Machine learning for differentiating lung squamous cell cancer
952
+ from adenocarcinoma using Clinical-Metabolic characteristics and 18F-FDG PET/CT
953
+ radiomics
954
+ remarks: null
955
+ software: PyRadiomics
956
+ year: 2024
957
+ Zhang2024B:
958
+ BrokenFeatures: 0
959
+ ClassBalance: 66
960
+ Dimensionality: 4.35
961
+ Missings: 0
962
+ authors: Zhang et al.
963
+ dataset: Zhang2024B
964
+ download_source: https://peerj.com/articles/17111/#supplemental-information
965
+ modality: CT
966
+ nFeatures: 833
967
+ nInstances: 192
968
+ outcome: Lymph node metastasis presence
969
+ pathology: Gastric cancer
970
+ publication_doi: https://doi.org/10.7717/peerj.17111
971
+ publication_title: 'Predictive nomogram for lymph node metastasis and survival in
972
+ gastric cancer using contrast-enhanced computed tomography-based radiomics: a
973
+ retrospective study'
974
+ remarks: null
975
+ software: PyRadiomics
976
+ year: 2024
@@ -0,0 +1,214 @@
1
+ # -*- coding: utf-8 -*-
2
+
3
+ """
4
+ radMLBench was developed at the University Hospital in Essen, Germany.
5
+ For questions use github issues or write an email (aydin.demircioglu@uk-essen.de).
6
+
7
+ radMLBench is partially based on the PMLB (Penn Machine Learning Benchmarks), see https://epistasislab.github.io/pmlb/
8
+
9
+ Permission is hereby granted, free of charge, to any person obtaining a copy of this software
10
+ and associated documentation files (the "Software"), to deal in the Software without restriction,
11
+ including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense,
12
+ and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so,
13
+ subject to the following conditions:
14
+
15
+ The above copyright notice and this permission notice shall be included in all copies or substantial
16
+ portions of the Software.
17
+
18
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT
19
+ LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
20
+ IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
21
+ WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
22
+ SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
23
+ """
24
+
25
+ import pandas as pd
26
+ import numpy as np
27
+ import os
28
+ from pkg_resources import resource_filename
29
+ import requests
30
+ import warnings
31
+ import subprocess
32
+ import pathlib
33
+ import yaml
34
+
35
+ from sklearn.model_selection import RepeatedStratifiedKFold
36
+ from sklearn.preprocessing import StandardScaler
37
+ import numpy as np
38
+
39
+ GITHUB_URL = 'https://github.com/aydindemircioglu/radMLBench/raw/main/datasets'
40
+ suffix = '.gz'
41
+ metaData = None
42
+
43
+
44
+
45
+ def listDatasets (sort_by = None):
46
+ """Return a list of all available datasets
47
+
48
+ Parameters
49
+ ----------
50
+ sort_by: str
51
+ The column to sort by. If column does not exist, the alphabetically sorted list will be returned.
52
+
53
+ Returns
54
+ ----------
55
+ datasets: list
56
+ The list of all available datasets, sorted by sort_by.
57
+ """
58
+ # ensure its loaded
59
+ global metaData
60
+ readMetaData()
61
+ dsets = list(metaData.keys())
62
+ try:
63
+ if sort_by is not None:
64
+ dsets = sorted(metaData.keys(), key=lambda x: metaData[x][sort_by])
65
+ except:
66
+ pass
67
+ return dsets
68
+
69
+
70
+
71
+ def get_dataset_url(GITHUB_URL, dataset_name, suffix):
72
+ dataset_url = '{GITHUB_URL}/{DATASET_NAME}{SUFFIX}'.format(
73
+ GITHUB_URL=GITHUB_URL,
74
+ DATASET_NAME=dataset_name,
75
+ SUFFIX=suffix
76
+ )
77
+
78
+ re = requests.get(dataset_url)
79
+ if re.status_code != 200:
80
+ raise ValueError(f'Unable to retrieve dataset from {dataset_url}')
81
+ return dataset_url
82
+
83
+
84
+
85
+ def loadData (dataset_name,return_X_y=False, local_cache_dir=None):
86
+ """Download a data set from the radMLBench, (optionally) store it locally, and return the data set.
87
+
88
+ You must be connected to the internet if you are fetching a data set that is not cached locally.
89
+
90
+ Parameters
91
+ ----------
92
+ dataset_name: str
93
+ The name of the data set to load from PMLB.
94
+ return_X_y: bool (default: False)
95
+ Whether to return the data in scikit-learn format, with the features
96
+ and labels stored in separate NumPy arrays.
97
+ local_cache_dir: str (default: None)
98
+ The directory on your local machine to store the data files.
99
+ If None, then the local data cache will not be used.
100
+
101
+ Returns
102
+ ----------
103
+ dataset: pd.DataFrame or (array-like, array-like)
104
+ if return_X_y == False: A pandas DataFrame containing the fetched data set (with columns ID and Target)
105
+ if return_X_y == True: A tuple of NumPy arrays containing (features, labels)
106
+ """
107
+
108
+ if dataset_name not in listDatasets():
109
+ raise ValueError('Dataset not found?')
110
+
111
+ if local_cache_dir is None:
112
+ dataset_url = get_dataset_url(GITHUB_URL, dataset_name, suffix)
113
+ dataset = pd.read_csv(dataset_url, compression='gzip')
114
+ else:
115
+ dataset_path = os.path.join(local_cache_dir, dataset_name+suffix)
116
+ if os.path.exists(dataset_path):
117
+ dataset = pd.read_csv(dataset_path, compression='gzip')
118
+ else:
119
+ print (f"Downloading to {dataset_path}...")
120
+ dataset_url = get_dataset_url(GITHUB_URL, dataset_name, suffix)
121
+ dataset = pd.read_csv(dataset_url, compression='gzip')
122
+ dataset_dir = os.path.split(dataset_path)[0]
123
+ if not os.path.isdir(dataset_dir):
124
+ os.makedirs(dataset_dir)
125
+ dataset.to_csv(dataset_path, compression='gzip', index=False)
126
+
127
+ if return_X_y:
128
+ X = dataset.drop(['ID', 'Target'], axis=1).values
129
+ y = dataset['Target'].values
130
+ return (X, y)
131
+ else:
132
+ return dataset
133
+
134
+
135
+
136
+ def getCVSplits(dataset, num_splits=10, num_repeats=10):
137
+ """
138
+ Generate indices for cross-validation splits.
139
+
140
+ Parameters:
141
+ - dataset (str or DataFrame or tuple of numpy arrays): Input dataset. If str, it will be treated as dataset name to load.
142
+ - num_splits (int): Number of folds for cross-validation.
143
+ - num_repeats (int): Number of times to repeat the cross-validation process.
144
+
145
+ Returns:
146
+ - List of tuples: Each tuple contains train and test indices for one split.
147
+ """
148
+ if isinstance(dataset, str):
149
+ dataset = loadDataset(dataset)
150
+
151
+ random_state = 42 * num_splits + num_repeats + 42
152
+
153
+ if isinstance(dataset, pd.DataFrame):
154
+ y = dataset["Target"].values
155
+ X = dataset.drop(columns=["Target", "ID"]).values
156
+ else:
157
+ X, y = dataset
158
+
159
+ cv_splits = []
160
+ rskf = RepeatedStratifiedKFold(n_splits=num_splits, n_repeats=num_repeats, random_state=random_state)
161
+ for train_index, test_index in rskf.split(X, y):
162
+ cv_splits.append((train_index, test_index))
163
+
164
+ return cv_splits
165
+
166
+
167
+
168
+ def readMetaData ():
169
+ """Internal. Reads metadata into variable"""
170
+ def read_yaml(file_path):
171
+ try:
172
+ with open(file_path, 'r') as stream:
173
+ data = yaml.safe_load(stream)
174
+ return data
175
+ except Exception as e:
176
+ print ('Unable to read metadata file for all dataset!')
177
+ print ('Error', e)
178
+ return None
179
+
180
+ global metaData
181
+ if metaData is None:
182
+ #print ("Reading metadata.")
183
+ package_dir = resource_filename('radMLBench', '')
184
+ metadata_path = os.path.join(package_dir, 'metadata.yaml')
185
+ metaData = read_yaml(metadata_path)
186
+ pass
187
+
188
+
189
+
190
+
191
+ def getMetaData (dataset_name):
192
+ """Retrieve the metadata for a data set from the radMLBench.
193
+
194
+ Parameters
195
+ ----------
196
+ dataset_name: str
197
+ The name of the data set to retrieve the metadata for.
198
+
199
+ Returns
200
+ ----------
201
+ metadata: dictionary
202
+ Dictionary containing the metadata
203
+ """
204
+ # ensure its loaded
205
+ global metaData
206
+ readMetaData()
207
+ if dataset_name not in metaData:
208
+ print ("Unknown dataset.")
209
+ return None
210
+ return metaData[dataset_name]
211
+
212
+
213
+
214
+ #
@@ -0,0 +1,21 @@
1
+ MIT License
2
+
3
+ Copyright (c) 2024, Aydin Demircioglu
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
@@ -0,0 +1,31 @@
1
+ Metadata-Version: 2.1
2
+ Name: radMLBench
3
+ Version: 1.0
4
+ Summary: A Python wrapper for the radMLBench data repository.
5
+ Home-page: https://github.com/aydindemircioglu/radMLBench
6
+ Author: Aydin Demircioglu
7
+ Author-email: aydin.demircioglu@uk-essen.de
8
+ License: MIT
9
+ Keywords: radiomics,data mining,benchmark,machine learning,data analysis,data sets,data science,wrapper
10
+ Classifier: Intended Audience :: Science/Research
11
+ Classifier: License :: OSI Approved :: MIT License
12
+ Classifier: Programming Language :: Python :: 3.7
13
+ Requires-Python: >=3.6.0
14
+ Description-Content-Type: text/markdown
15
+ License-File: LICENSE.txt
16
+ Requires-Dist: pandas (>=1.2.0)
17
+ Requires-Dist: requests (>=2.18.0)
18
+ Requires-Dist: pyyaml (>=5.2)
19
+ Requires-Dist: joblib (>=0.12.0)
20
+ Requires-Dist: numpy (>=1.18.0)
21
+ Requires-Dist: scikit-learn (>=0.20.0)
22
+
23
+
24
+ A Python wrapper for the radMLBench data repository.
25
+
26
+ Contact
27
+ =============
28
+ If you have any questions or comments about radMLBench,
29
+ please feel free to contact us via e-mail: aydin.demircioglu@uk-essen.de
30
+
31
+ This project is hosted at https://github.com/aydindemircioglu/radMLBench
@@ -0,0 +1,9 @@
1
+ radMLBench/__init__.py,sha256=yqNSt_MuyFATNDFfO_TQ-Jvb_BIQheRg4M7188y7Q-A,1428
2
+ radMLBench/metadata.yaml,sha256=c5-eKbcXZV4-sHHyIhwmbqSqG0p5DbzcwxmvG_zJPqc,32004
3
+ radMLBench/radMLBench.py,sha256=H9iw7f-38J2se0zA36rUOvvFjWlErQg4rzThYkrKJuM,7127
4
+ radMLBench-1.0.dist-info/LICENSE.txt,sha256=gTSULos27L9Ns-K-6CiVNHN6JNSQKEejyUh5fmuq_E8,1075
5
+ radMLBench-1.0.dist-info/METADATA,sha256=pJYZyOUfqQ3NbWD1p8YXFEdTzLQ5oQM-vc9e1vY5muA,1083
6
+ radMLBench-1.0.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
7
+ radMLBench-1.0.dist-info/top_level.txt,sha256=5BKhbStaBdsf74i1jicPCX82ERw2V9Ww2tHjAq7sZ48,11
8
+ radMLBench-1.0.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
9
+ radMLBench-1.0.dist-info/RECORD,,
@@ -0,0 +1,5 @@
1
+ Wheel-Version: 1.0
2
+ Generator: bdist_wheel (0.40.0)
3
+ Root-Is-Purelib: true
4
+ Tag: py3-none-any
5
+
@@ -0,0 +1 @@
1
+ radMLBench
@@ -0,0 +1 @@
1
+