<jupyter_start><jupyter_text>Working on Model<jupyter_code>import torch
import torch.nn as nn
import pandas as pd
import numpy as np
import os<jupyter_output><empty_output><jupyter_text>Approach Based on the Following<jupyter_code># https://github.com/AndriyMulyar/bert_document_classification<jupyter_output><empty_output><jupyter_text>Unfortunately as shown in the following lines, this repo has problems!!!!<jupyter_code>from bert_document_classification.models import SmokerPhenotypingBert
from bert_document_classification.models import ObesityPhenotypingBert
smoking_classifier = SmokerPhenotypingBert(device='cuda', batch_size=10) #defaults to GPU prediction|

smoking_classifier = SmokerPhenotypingBert(device='cpu', batch_size=10) #defaults to GPU prediction
obesity_classifier = ObesityPhenotypingBert(device='cpu', batch_size=10) #or CPU if you would like.
smoking_classifier.predict(["I'm a document! Make me long and the model can still perform well!"])<jupyter_output><empty_output>
{'fraction_non_alphanumeric': 0.08032128514056225, 'fraction_numerical': 0.006024096385542169, 'mean_word_length': 10.077777777777778, 'pattern_counts': {'":': 0, '<': 13, '<?xml version=': 0, '>': 13, 'https://': 1, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'chain_length': '3', 'dir': 'jupyter-structured-clean-dedup', 'id': '591377', 'license': 'MIT', 'n_tokens_mistral': 334, 'n_tokens_neox': 305, 'n_words': 76, 'path': 'notebooks/6.0-pb-TryingGithubRepoMultiLabel_63stars.ipynb', 'repo_name': 'WisConT/oecd_financial_flows', 'size': '996'}