chunked content to 512
This commit is contained in:
parent
3864d7e93c
commit
679f1bd941
@ -44,16 +44,24 @@ if __name__ == "__main__":
|
|||||||
|
|
||||||
Base.metadata.create_all(engine)
|
Base.metadata.create_all(engine)
|
||||||
|
|
||||||
result = analyzer.classify_article_finbert("Strong earning growth and expending market shares have positionned the company for long term success.")
|
# result = analyzer.classify_article_finbert("Strong earning growth and expending market shares have positionned the company for long term success.")
|
||||||
print(f'result {result}')
|
|
||||||
|
|
||||||
|
|
||||||
# for file, content in html_files.items():
|
|
||||||
# result = analyzer.classify_article_finbert(content)
|
|
||||||
|
|
||||||
# filename = os.path.basename(file)
|
|
||||||
# print(f'result {result}')
|
# print(f'result {result}')
|
||||||
|
|
||||||
|
for file, content in html_files.items():
|
||||||
|
chunk_size = 512
|
||||||
|
chunks = [content[i:i+chunk_size] for i in range(0, len(content), chunk_size)]
|
||||||
|
|
||||||
|
results = []
|
||||||
|
for chunk in chunks:
|
||||||
|
if chunk.strip():
|
||||||
|
chunk_result = analyzer.classify_article_finbert(chunk)
|
||||||
|
results.extend(chunk_result)
|
||||||
|
|
||||||
|
result = results if results else [{'label': 'neutral', 'score': 0.0}]
|
||||||
|
|
||||||
|
filename = os.path.basename(file)
|
||||||
|
print(f'result {result}')
|
||||||
|
|
||||||
# label = result[0]['label']
|
# label = result[0]['label']
|
||||||
# score = result[0]['score']
|
# score = result[0]['score']
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user