Skip to content
This repository was archived by the owner on Mar 8, 2024. It is now read-only.

Commit 5d5bb1b

Browse files
authored
adds default arguments for CRFTagger in AL extraction code lookup (#56)
1 parent 44aae35 commit 5d5bb1b

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

src/app/weak-supervision/components/information-sources-code-lookup.ts

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,13 @@ class ATLClassifier(LearningClassifier):
6161
class ATLExtractor(LearningExtractor):
6262
6363
def __init__(self):
64-
self.model = CRFTagger()
64+
self.model = CRFTagger(
65+
num_epochs = 100, # Number of epochs to train the CRF tagger
66+
learning_rate = 0.001, # Factor to apply during backpropagation
67+
momentum = 0.9, # Factor to weigh previous iteration during training
68+
random_seed = None, # Random seed to use for reproducibility. If None, a random seed is chosen
69+
verbose = False, # set to True to see the training progress
70+
)
6571
6672
@params_fit(
6773
embedding_name = "@@EMBEDDING@@", # pick this from the options above

0 commit comments

Comments
 (0)