From the uDemy course on LLM engineering.
https://www.udemy.com/course/llm-engineering-master-ai-and-large-language-models
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
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]) |