15 lines
468 B
Python
15 lines
468 B
Python
|
# Importowanie potrzebnych bibliotek
|
||
|
|
import pickle
|
||
|
|
import pandas as pd
|
||
|
|
|
||
|
|
#Nazwa pliku z modelem
|
||
|
|
filename = "flat_model.pkl"
|
||
|
|
|
||
|
|
#Ładowanie modelu
|
||
|
|
model = pickle.load(open(filename, "rb"))
|
||
|
|
|
||
|
|
#Definicja funkcji predykcyjnej
|
||
|
|
def predict(stan, czynsz, liczba_pokoi, metraz, rynek):
|
||
|
|
params = pd.DataFrame.from_dict({'stan': [stan], 'czynsz': [czynsz], 'liczba pokoi': [liczba_pokoi], 'metraz': [metraz], 'rynek': [rynek]})
|
||
|
|
return model.predict(params).item()
|