Commit aa3f1381 authored by Janos Borst's avatar Janos Borst
Browse files

Make Apex models optional

parent 5e4d8b18
Pipeline #35955 passed with stages
in 2 minutes and 16 seconds
......@@ -15,13 +15,17 @@ try:
from .ZAGCNN import ZAGCNN
from .SKG_ML import SKG
except:
print("pytorch_geometric not installed.")
pass
#Experimental Models
from .ConceptScores import ConceptScores, ConceptScoresCNN,ConceptScoresCNNAttention,KimCNN2Branch,ConceptProjection,ConceptScoresAttention, ConceptScoresRelevance, ConceptScoresRelevanceWithImportanceWeights
from .ConceptLSAN import ConceptLSAN
from .GloveAsConcept import GloveConcepts
from .BertAsConcept import BertAsConcept, BertAsConcept2, BertAsConcept3
from .BertAsConcept2 import BertAsConceptFineTuning, BertAsConceptCLSFinetuning
from .TwoLanguageModels import LMVSLM_Classifier
try:
from .BertAsConcept import BertAsConcept, BertAsConcept2, BertAsConcept3
from .BertAsConcept2 import BertAsConceptFineTuning, BertAsConceptCLSFinetuning
from .TwoLanguageModels import LMVSLM_Classifier
except:
print("Apex not installed.")
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment