lsha49 / FairBERT

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

FAIR BERT PROJECT

This is a casenote script file containing model training and testing using LEGAL BERT model published in huggingface

Requirements

  • The project is developed using the latest version of tensorflow (wheel is for Mac M1)
  • The most critical dependency is HuggingFace, python3

Interface for input

During development, the only input file needed is a csv file containing:

  • col1: Sentence
  • col2-n: the six label categories of the sentence

How to run this?

First add a data.csv file under data/ folder, then a test run is as simple as

python3 tfBertClf.py

About


Languages

Language:Python 100.0%