test commit

This commit is contained in:
Andrea Pedrotti 2023-02-09 16:47:17 +01:00
parent 8325262972
commit 4485d97e03
2 changed files with 3 additions and 3 deletions

View File

@ -9,7 +9,7 @@ import numpy as np
from vgfs.commons import TfidfVectorizerMultilingual
from vgfs.learners.svms import MetaClassifier, get_learner
from vgfs.multilingualGen import MultilingualGen
from vgfs.transformerGen import TransformerGen
from gfun.vgfs.textualTransformerGen import TextualTransformerGen
from vgfs.vanillaFun import VanillaFunGen
from vgfs.wceGen import WceGen
@ -98,7 +98,7 @@ class GeneralizedFunnelling:
self.first_tier_learners.append(wce_vgf)
if self.trasformer_vgf:
transformer_vgf = TransformerGen(
transformer_vgf = TextualTransformerGen(
model_name=self.transformer_name,
lr=self.lr_transformer,
epochs=self.epochs,

View File

@ -21,7 +21,7 @@ transformers.logging.set_verbosity_error()
# TODO: add support to loggers
class TransformerGen:
class TextualTransformerGen:
def __init__(
self,
model_name,