hodorfi commited on
Commit
fd2334d
Β·
1 Parent(s): d37eb5e

Update pages/4_πŸ”¬_Decision_Exploration_Panel.py

Browse files
pages/4_πŸ”¬_Decision_Exploration_Panel.py CHANGED
@@ -152,8 +152,10 @@ def get_model(model_path):
152
 
153
  @st.cache(allow_output_mutation=True)
154
  def get_feature_vector_model(keras_model):
155
- feature_mod = get_feature_extractor_model(keras_model)
156
- return feature_mod
 
 
157
 
158
  def load_pd_data_frame(df_csv_path):
159
  return pd.read_csv(df_csv_path)
 
152
 
153
  @st.cache(allow_output_mutation=True)
154
  def get_feature_vector_model(keras_model):
155
+ feature_extractor = tf.keras.Model(keras_model.inputs,keras_model.layers[-3].output)
156
+ return feature_extractor
157
+ # feature_mod = get_feature_extractor_model(keras_model)
158
+ # return feature_mod
159
 
160
  def load_pd_data_frame(df_csv_path):
161
  return pd.read_csv(df_csv_path)