Doctorado-ML/fimdlp
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/FImdlp.git synced 2025-08-17 16:35:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: ⚡

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2022-12-21 19:05:24 +01:00
parent 9db16d9d3c
commit 2775698063
10 changed files with 71 additions and 209 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
samples/sample.py
View File

@@ -14,10 +14,8 @@ datasets = {
}
ap = argparse.ArgumentParser()
ap.add_argument("--proposal", action="store_const", const=1)
ap.add_argument("--original", dest="proposal", action="store_const", const=0)
ap.add_argument(
"--alternative", dest="proposal", action="store_const", const=2
"--alternative", dest="proposal", action="store_const", const=1
)
ap.add_argument("dataset", type=str, choices=datasets.keys())
args = ap.parse_args()
@@ -32,7 +30,7 @@ class_name = df.columns.to_list()[class_column]
X = df.drop(class_name, axis=1)
y, _ = pd.factorize(df[class_name])
X = X.to_numpy()
test = FImdlp(proposal=args.proposal)
test = FImdlp(algorithm=args.proposal if args.proposal is not None else 0)
now = time.time()
test.fit(X, y)
fit_time = time.time()