From 9802fe936a4a1fdb3c34cc29918fbec2f4cd9d4a Mon Sep 17 00:00:00 2001 From: bav1g20 <bav1g20@soton.ac.uk> Date: Wed, 7 Jun 2023 16:58:37 +0200 Subject: [PATCH] updated data saving as spec was not clear --- src/main.py | 40 +++++++++++++++++++++++++++++++++++----- 1 file changed, 35 insertions(+), 5 deletions(-) diff --git a/src/main.py b/src/main.py index a607170..a027605 100644 --- a/src/main.py +++ b/src/main.py @@ -27,6 +27,39 @@ def handleDataset(ds): handleTrain(ds) elif i =='a': handlePredict(ds) + elif i=='c': + internal(ds) + +def internal(ds): + print(f"Predicting {ds} with model") + + root = Path(__file__).parent.parent + path = root / "models" / datasets[ds] + + files = path.glob('*') + + print("Available predictors: ") + for i,val in enumerate(files): + print(f"[{i:2<}] {val}") + print("[q] Quit") + + i = input() + if i == 'q': + exit() + + root = Path(__file__).parent.parent + testing = root / "resources" / "TestingData{}.csv".format("Multi" if ds =='b' else "Binary") + training = root / "resources" / "TrainingData{}.csv".format("Multi" if ds =='b' else "Binary") + testingData = read_csv(testing, header=None) + trainingData = read_csv(training, header=None) + + x = list(path.glob('*')) + f = x[int(i)] + predictor = TabularPredictor.load(path=f.absolute().as_posix()) + + n = predictor.leaderboard() + print(n["model"].values.tolist()) + def handlePredict(ds): print(f"Predicting {ds} with model") @@ -47,7 +80,6 @@ def handlePredict(ds): try: i = int(i) predict(ds, list(path.glob('*'))[i]) - except Exception: print() @@ -66,9 +98,7 @@ def predict(ds, model): savePath = savePath.absolute().as_posix() print(f"Saving to {savePath}") - with open(savePath, "w") as f: - for i in a: - f.write(f"{i}\n") + testingData.to_csv(savePath, header=False, index=False) def handleTrain(ds): print("Starting training") @@ -85,7 +115,7 @@ def handleTrain(ds): eval_metric="accuracy", path=file.absolute().as_posix() ) - predictor.fit(train_data) + predictor.fit(train_data,verbosity=4) print("Finished predictor!") -- GitLab