Configs

Predefined topic labelling model configurations, located at atnlp/share/config

atnlp/share/config/logreg.yml
1
2
3
4
5
6
model: logreg
params:
  tfidf__stop_words: english
  tfidf__min_df: 2
  tfidf__max_df: 0.9
  logreg__estimator__C: 30.0
atnlp/share/config/svm.yml
1
2
3
4
5
6
7
model: svm
params:
  tfidf__stop_words: english
  tfidf__min_df: 2
  tfidf__max_df: 0.9
  svc__estimator__kernel: linear
  svc__estimator__C: 3.0
atnlp/share/config/wordmatch.yml
1
2
3
4
5
model: wordmatch
params:
  bow__stop_words: english
  bow__min_df: 2
  bow__max_df: 0.9