From 12979028957708bd42fb1dd6bf60264da9f9600b Mon Sep 17 00:00:00 2001 From: Lorenzo Volpi Date: Tue, 12 Sep 2023 17:41:28 +0200 Subject: [PATCH] Refactoring --- quacc/main.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/quacc/main.py b/quacc/main.py index f8e14aa..d28b9ae 100644 --- a/quacc/main.py +++ b/quacc/main.py @@ -9,7 +9,7 @@ from quacc.estimator import ( MulticlassAccuracyEstimator, ) -from quacc.data import get_dataset +from quacc.dataset import getImdbTrainTest qp.environ["SAMPLE_SIZE"] = 100 @@ -20,7 +20,7 @@ dataset_name = "imdb" def estimate_multiclass(): print(dataset_name) - train, test = get_dataset(dataset_name) + train, test = getImdbTrainTest(dataset_name) model = LogisticRegression() @@ -59,7 +59,7 @@ def estimate_multiclass(): def estimate_binary(): print(dataset_name) - train, test = get_dataset(dataset_name) + train, test = getImdbTrainTest(dataset_name) model = LogisticRegression()