Compare commits
2 Commits
ef4fd624e4
...
885f51b83d
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
885f51b83d | ||
|
|
4f665db568 |
4
.gitignore
vendored
4
.gitignore
vendored
@ -3,8 +3,8 @@
|
||||
.idea/*
|
||||
|
||||
data/
|
||||
.zip
|
||||
.db
|
||||
*.zip
|
||||
*.db
|
||||
plots/
|
||||
results/
|
||||
|
||||
|
||||
@ -74,5 +74,12 @@ class ArticleAnalyzer:
|
||||
|
||||
def classify_article_finbert(self, article_html):
|
||||
article_md = self.convert_to_markdown(article_html)
|
||||
result = self.classifier(article_md)
|
||||
return result
|
||||
chunk_size = 512
|
||||
chunks = [article_md[i:i + chunk_size] for i in range(0, len(article_md), chunk_size)]
|
||||
|
||||
results = []
|
||||
for chunk in chunks:
|
||||
result = self.classifier(chunk)
|
||||
results.append(result)
|
||||
|
||||
return results
|
||||
Loading…
x
Reference in New Issue
Block a user