def MakeTable(): print(args.csvname) table = datasets.LoadDmv(args.csvname) # modify oracle_est = estimators_lib.Oracle(table) if args.run_bn: return table, common.TableDataset(table), oracle_est return table, None, oracle_est
def MakeTable(): assert args.dataset in ['dmv-tiny', 'dmv'] if args.dataset == 'dmv-tiny': table = datasets.LoadDmv('dmv-tiny.csv') elif args.dataset == 'dmv': table = datasets.LoadDmv() oracle_est = estimators_lib.Oracle(table) if args.run_bn: return table, common.TableDataset(table), oracle_est return table, None, oracle_est
def MakeTable(): if args.dataset == 'dmv-tiny': table = datasets.LoadDmv('dmv-tiny.csv') elif args.dataset == 'dmv': table = datasets.LoadDmv(args.version + '.csv') # modify elif args.dataset == 'PF': table = datasets.LoadPF(args.version + '.csv') # modify oracle_est = estimators_lib.Oracle(table) if args.run_bn: return table, common.TableDataset(table), oracle_est return table, None, oracle_est