search

ppapalampidi / SUMMER

Screenplay Summarization using Latent Narrative Structure
35 stars 9 forks source link

Fix for updated SKLearn method signature #4

Open vade opened 2 years ago

vade commented 2 years ago

Seems like SKLearn got picky about function arguments. Following this advice fixed the issue and im now able to train SUMMER :)

https://stackoverflow.com/questions/69783897/compute-class-weight-function-issue-in-sklearn-library-when-used-in-keras-cl

Thank you!