matsammut commited on
Commit
221bc87
·
verified ·
1 Parent(s): dab8a0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -65,8 +65,14 @@ def cleaning_features(data):
65
 
66
 
67
  def pca(data):
68
- encoder = joblib.load('onehot_encoder.joblib')
69
- pca_model = joblib.load('pca.joblib')
 
 
 
 
 
 
70
  one_hot_encoded = encoder.transform(data[['workclass', 'occupation']])
71
  encoded_columns_df = pd.DataFrame(one_hot_encoded, columns=encoder.get_feature_names_out())
72
  pca_result_net = pca_model.transform(encoded_columns_df)
 
65
 
66
 
67
  def pca(data):
68
+ encoder_pkl = 'onehot_encoder.pkl'
69
+ pca_model_pkl = 'pca.pkl'
70
+
71
+ with open(pca_model_pkl, 'rb') as file:
72
+ pca_model = pickle.load(pca_model_pkl)
73
+ with open(encoder_pkl, 'rb') as file:
74
+ encoder = pickle.load(encoder_pkl)
75
+
76
  one_hot_encoded = encoder.transform(data[['workclass', 'occupation']])
77
  encoded_columns_df = pd.DataFrame(one_hot_encoded, columns=encoder.get_feature_names_out())
78
  pca_result_net = pca_model.transform(encoded_columns_df)