mlf1_setup.py

import os
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline

from sklearn.model_selection import train_test_split
from sklearn.metrics import f1_score, accuracy_score

import torch
from simpletransformers.classification import ClassificationModel


# [Optional] Releases all unoccupied cached memory currently held by the 
# caching allocator so that those can be used in other GPU application
torch.cuda.empty_cache()