117 rulebased model #130

yiwen-h · 2023-08-03T11:26:33Z

can now add rules to boost model prediction performance - if certain words appear in text, this increases probability of specific classes
words/rules are defined in rules_dict, found in params.py
haven't added tests for the new functionality however

Closes #117

yiwen-h added 8 commits July 27, 2023 10:32

testing increasing prob of class if certain words are present in text

525f754

moved rules to params for more flexibility

72d673a

added rules to pipeline. testing with test_rules.py

f62ba5b

finished rules_dict (subject to tweaking)

0902caa

added rules to bert predictions

f0a023e

get_multilabel_metrics now uses predict_multilabel_bert

cca07bd

added probs_dict for variable probabilities too

669fdf0

moved test_rules out of tests folder, reran coverage

de20e3a

yiwen-h requested a review from ChrisBeeley August 3, 2023 11:26

yiwen-h self-assigned this Aug 3, 2023

Merge branch 'development' into 117_rulebased_model

b3d2969

yiwen-h merged commit 0d366e9 into development Aug 4, 2023
3 checks passed

yiwen-h deleted the 117_rulebased_model branch August 4, 2023 09:03

Provide feedback