defaults.py 839 B

12345678910111213141516171819202122232425
  1. # In this file we have defined the default values for IEPY settings
  2. # These default values will be always used except explicitely said.
  3. from iepy.utils import make_feature_list
  4. extractor_config = {
  5. "classifier": "svc",
  6. "classifier_args": {},
  7. "dense_features": make_feature_list("""
  8. entity_order
  9. entity_distance
  10. other_entities_in_between
  11. verbs_count_in_between
  12. verbs_count
  13. total_number_of_entities
  14. symbols_in_between
  15. number_of_tokens
  16. """),
  17. "sparse_features": make_feature_list("""
  18. bag_of_words
  19. bag_of_pos
  20. bag_of_words_in_between
  21. bag_of_pos_in_between
  22. """)
  23. }