Compare commits
2 Commits
ef4fd624e4
...
885f51b83d
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
885f51b83d | ||
|
|
4f665db568 |
4
.gitignore
vendored
4
.gitignore
vendored
@ -3,8 +3,8 @@
|
|||||||
.idea/*
|
.idea/*
|
||||||
|
|
||||||
data/
|
data/
|
||||||
.zip
|
*.zip
|
||||||
.db
|
*.db
|
||||||
plots/
|
plots/
|
||||||
results/
|
results/
|
||||||
|
|
||||||
|
|||||||
@ -74,5 +74,12 @@ class ArticleAnalyzer:
|
|||||||
|
|
||||||
def classify_article_finbert(self, article_html):
|
def classify_article_finbert(self, article_html):
|
||||||
article_md = self.convert_to_markdown(article_html)
|
article_md = self.convert_to_markdown(article_html)
|
||||||
result = self.classifier(article_md)
|
chunk_size = 512
|
||||||
return result
|
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