main.py

from utils import getCifar10Data, getMnistData, getFMnistData, saveAsPickle,getTinyImagenetData
from algorithms.ghcidr import GHCIDR
from algorithms.baselineRHC import RHC
from algorithms.koncw import KONCW 
from algorithms.cwkc import CWKC
from algorithms.rhckon import RHCKON
import argparse
import os,sys

parser = argparse.ArgumentParser()
parser.add_argument("-variantName", default='MergeGHCIDR', type=str)
parser.add_argument("-datasetName", default='MNIST', type=str)
parser.add_argument("-storeClusters", default=True, type=bool)
parser.add_argument("-savePicklePath", default='../reducedData/')
parser.add_argument("-alpha", default=0.3, type=float) 
parser.add_argument("-KFarthest", default=1, type=int)
args = parser.parse_args()

Datasets = {
            "MNIST":getMnistData,
            "FMNIST":getFMnistData, 
            "CIFAR10":getCifar10Data,
            "TinyImagenet":getTinyImagenetData
            }
Variants = {
            "GHCIDR":GHCIDR,
            "KONCW":KONCW,
            "CWKC":CWKC,
            "RHC":RHC,
            "RHCKON":RHCKON
            }

dataCall, variantCall = Datasets[args.datasetName],Variants[args.variantName]
reducedData = variantCall(dataCall(),args)
path =  "./datasetPickle/" +args.datasetName + '_' + args.variantName +".pickle"
saveAsPickle(reducedData,path)