andrea.pedrotti
/
gfun_multimodal
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
gfun/generalizedFunnelling.py
View File

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

2
gfun/vgfs/transformerGen.py → gfun/vgfs/textualTransformerGen.py
View File

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