ttd22 commited on
Commit
f997324
1 Parent(s): deb98a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import numpy as np
8
  import pandas as pd
9
  import matplotlib.pyplot as plt
10
 
11
-
12
  def load_data(selected_columns):
13
  dataset = load_dataset("ttd22/house-price", streaming = True)
14
  df = pd.DataFrame.from_dict(dataset["train"])
@@ -17,6 +17,7 @@ def load_data(selected_columns):
17
  X, y = df.drop("SalePrice", axis=1), df["SalePrice"]
18
  return X,y
19
 
 
20
  def load_model(X, y):
21
  X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
22
  params = {'n_estimators': 378, 'num_leaves': 31, 'max_depth': 35, 'learning_rate': 0.01713200441531346, 'colsample_bytree': 0.5167187468778591, 'subsample': 0.7380799250244564, 'lambda_l1': 4.712815578969268, 'lambda_l2': 1.7427073600749474, 'min_child_weight': 9.962946763677316}
@@ -32,6 +33,7 @@ selected_columns = ['OverallQual', 'GrLivArea', 'TotalBsmtSF', 'YearBuilt', 'Bsm
32
  X,y = load_data(selected_columns)
33
  model = load_model(X, y)
34
 
 
35
  def buil_UI(X):
36
  # min_values = X.min()
37
  # max_values = X.max()
 
8
  import pandas as pd
9
  import matplotlib.pyplot as plt
10
 
11
+ @st.experimental_memo
12
  def load_data(selected_columns):
13
  dataset = load_dataset("ttd22/house-price", streaming = True)
14
  df = pd.DataFrame.from_dict(dataset["train"])
 
17
  X, y = df.drop("SalePrice", axis=1), df["SalePrice"]
18
  return X,y
19
 
20
+ @st.experimental_memo
21
  def load_model(X, y):
22
  X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
23
  params = {'n_estimators': 378, 'num_leaves': 31, 'max_depth': 35, 'learning_rate': 0.01713200441531346, 'colsample_bytree': 0.5167187468778591, 'subsample': 0.7380799250244564, 'lambda_l1': 4.712815578969268, 'lambda_l2': 1.7427073600749474, 'min_child_weight': 9.962946763677316}
 
33
  X,y = load_data(selected_columns)
34
  model = load_model(X, y)
35
 
36
+ @st.experimental_memo
37
  def buil_UI(X):
38
  # min_values = X.min()
39
  # max_values = X.max()