lorenzo.volpi
/
QuAcc
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactoring

This commit is contained in:
Lorenzo Volpi 2023-09-12 17:41:28 +02:00
parent 2981910b62
commit 1297902895
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
quacc/main.py
View File

@ -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()