From 4749a49bfd7fe6db0553ab4f6ab4bb21d3aa6322 Mon Sep 17 00:00:00 2001 From: felix0496 Date: Wed, 14 Sep 2022 14:01:47 +0200 Subject: [PATCH] adds default arguments for CRFTagger in AL extraction code lookup --- .../components/information-sources-code-lookup.ts | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/app/weak-supervision/components/information-sources-code-lookup.ts b/src/app/weak-supervision/components/information-sources-code-lookup.ts index 52d06e8c..f052abca 100644 --- a/src/app/weak-supervision/components/information-sources-code-lookup.ts +++ b/src/app/weak-supervision/components/information-sources-code-lookup.ts @@ -61,7 +61,13 @@ class ATLClassifier(LearningClassifier): class ATLExtractor(LearningExtractor): def __init__(self): - self.model = CRFTagger() + self.model = CRFTagger( + num_epochs = 100, # Number of epochs to train the CRF tagger + learning_rate = 0.001, # Factor to apply during backpropagation + momentum = 0.9, # Factor to weigh previous iteration during training + random_seed = None, # Random seed to use for reproducibility. If None, a random seed is chosen + verbose = False, # set to True to see the training progress + ) @params_fit( embedding_name = "@@EMBEDDING@@", # pick this from the options above