Speed up class-oriented narrative models

relatio-nlp / relatio

code base for constructing narrative statements from text

MIT License

93 stars 26 forks source link

Closed PinchOfData closed 2 years ago

PinchOfData commented 2 years ago

Gains of speed are to be found for

NarrativeModel.train(): cdist() is a costly operation, perhaps call it by batches rather than one phrase at a time
NarrativeModel.predict(): cdist() is a costly operation, perhaps call it by batches rather than one phrase at a time