bert_list_pred_import.py

from . import run_classifier
import os
import tensorflow as tf
import json

def getListPrediction(in_sentences):
    #1
    input_examples = [run_classifier.InputExample(guid="", text_a = x, text_b = None, label = "0") for x in in_sentences] # here, "" is just a dummy label
    
    #2
    input_features = run_classifier.convert_examples_to_features(input_examples, label_list, MAX_SEQ_LENGTH, tokenizer)
    
    #3
    predict_input_fn = run_classifier.input_fn_builder(features=input_features, seq_length=MAX_SEQ_LENGTH, is_training=False, drop_remainder=False)
    
    #4
    predictions = estimator.predict(input_fn=predict_input_fn)
    
    return predictions