Skip to content

Instantly share code, notes, and snippets.

@prateekjoshi565

prateekjoshi565/import_lib_bert.py

Last active July 17, 2020 13:34
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save prateekjoshi565/48e2a18eb402cd98ee7b916145b52d67 to your computer and use it in GitHub Desktop.
Save prateekjoshi565/48e2a18eb402cd98ee7b916145b52d67 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
import_lib_bert.py
import numpy as np
import pandas as pd
import torch
import torch.nn as nn
from sklearn.model_selection import train_test_split
from sklearn.metrics import classification_report
import transformers
from transformers import AutoModel, BertTokenizerFast
# specify GPU
device = torch.device("cuda")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment