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()