Legal Aircraft Lease Agreement Clause Binary Classifier (Bert)

Description

The legclf_aircraft_lease_agreement_bert model is a Bert Sentence Embeddings Document Classifier used to classify if the document belongs to the class aircraft-lease-agreement or not (Binary Classification).

Unlike the Longformer model, this model is lighter in terms of inference time.

Predicted Entities

aircraft-lease-agreement, other

Download Copy S3 URI

How to use

 
document_assembler = nlp.DocumentAssembler()\
    .setInputCol("text")\
    .setOutputCol("document")
  
embeddings = nlp.BertSentenceEmbeddings.pretrained("sent_bert_base_cased", "en")\
    .setInputCols("document")\
    .setOutputCol("sentence_embeddings")
    
doc_classifier = legal.ClassifierDLModel.pretrained("legclf_aircraft_lease_agreement_bert", "en", "legal/models")\
    .setInputCols(["sentence_embeddings"])\
    .setOutputCol("category")
    
nlpPipeline = nlp.Pipeline(stages=[
    document_assembler, 
    embeddings,
    doc_classifier])
 
df = spark.createDataFrame([["YOUR TEXT HERE"]]).toDF("text")

model = nlpPipeline.fit(df)

result = model.transform(df)

Model Information

Model Name:	legclf_aircraft_lease_agreement_bert
Compatibility:	Legal NLP 1.0.0+
License:	Licensed
Edition:	Official
Input Labels:	[sentence_embeddings]
Output Labels:	[class]
Language:	en
Size:	22.2 MB

References

Legal documents, scrapped from the Internet, and classified in-house

Benchmarking

                   label  precision    recall  f1-score   support
aircraft-lease-agreement       0.94      0.91      0.92        33
                   other       0.95      0.97      0.96        59
                accuracy          -         -      0.95        92
               macro-avg       0.94      0.94      0.94        92
            weighted-avg       0.95      0.95      0.95        92

PREVIOUSLegal Administrative Services Agreement Clause Binary Classifier (Longformer)

NEXTLegal Aircraft Lease Agreement Clause Binary Classifier (Longformer)