Mr-Thop commited on
Commit
0dc5151
·
1 Parent(s): fe50a9e

Load Scaler and encoders

Browse files
Files changed (1) hide show
  1. load_models.py +9 -0
load_models.py CHANGED
@@ -1,6 +1,15 @@
1
  from huggingface_hub import hf_hub_download
2
  import joblib
3
 
 
 
 
 
 
 
 
 
 
4
  def load_rf():
5
  model = joblib.load(
6
  hf_hub_download("Mr-Thop/SCAAI", "rf.joblib")
 
1
  from huggingface_hub import hf_hub_download
2
  import joblib
3
 
4
+ def load_req():
5
+ scaler = joblib.load(
6
+ hf_hub_download("Mr-Thop/SCAAI", "scaler.joblib")
7
+ )
8
+ encoder = joblib.load(
9
+ hf_hub_download("Mr-Thop/SCAAI", "encoder.joblib")
10
+ )
11
+ return scaler, encoder
12
+
13
  def load_rf():
14
  model = joblib.load(
15
  hf_hub_download("Mr-Thop/SCAAI", "rf.joblib")