mirror of
https://github.com/Doctorado-ML/Stree_datasets.git
synced 2025-08-15 07:26:02 +00:00
28 lines
736 B
Python
28 lines
736 B
Python
import sys
|
|
import numpy as np
|
|
from experimentation.Sets import Datasets
|
|
|
|
set_name = "aaai"
|
|
if len(sys.argv) > 1:
|
|
set_name = sys.argv[1]
|
|
if set_name != "aaai" and set_name != "tanveer":
|
|
print("First parameter has to be one of: {aaai, tanveer}")
|
|
exit(1)
|
|
if len(sys.argv) > 2:
|
|
csv = sys.argv[2] == "tex"
|
|
else:
|
|
csv = False
|
|
|
|
datasets = Datasets(False, False, set_name)
|
|
if csv:
|
|
for number, dataset in enumerate(datasets):
|
|
X, y = datasets.load(dataset[0]) # type: ignore
|
|
samples, features = X.shape
|
|
classes = len(np.unique(y))
|
|
print(
|
|
"%d & %s & %d & %d & %d \\\\"
|
|
% (number + 1, dataset[0], X.shape[0], X.shape[1], classes)
|
|
)
|
|
else:
|
|
datasets.report()
|