You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

18 lines
479 B

# imports
import os
import re
from typing import List
from sentence_transformers import SentenceTransformer
import joblib
class RandomForestAgent:
def __init__(self):
self.vectorizer = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
self.model = joblib.load('random_forest_model.pkl')
def price(self, description: str) -> float:
vector = self.vectorizer.encode([description])
return max(0, self.model.predict(vector)[0])