13 lines
378 B
Python
13 lines
378 B
Python
|
import numpy as np
|
||
|
|
import pandas as pd
|
||
|
|
|
||
|
|
from madewithml import predict
|
||
|
|
|
||
|
|
|
||
|
|
def get_label(text, predictor):
|
||
|
|
df = pd.DataFrame({"title": [text], "description": "", "tag": "other"})
|
||
|
|
z = predictor.predict(data=df)["predictions"]
|
||
|
|
preprocessor = predictor.get_preprocessor()
|
||
|
|
label = predict.decode(np.stack(z).argmax(1), preprocessor.index_to_class)[0]
|
||
|
|
return label
|