bdpedigo commited on
Commit
44ce765
·
verified ·
1 Parent(s): 830be14

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. train.py +5 -5
train.py CHANGED
@@ -37,7 +37,7 @@ label_df = label_df.rename(columns=lambda x: x.replace(".1", ""))
37
 
38
  # %%
39
 
40
- X_df = pd.read_csv(out_path / model_name / "features.csv", index_col=[0, 1])
41
 
42
 
43
  # %%
@@ -300,9 +300,9 @@ report_by_class.index.name = "class"
300
  report_by_class.reset_index(inplace=True)
301
  # %%
302
 
303
- model_pickle_file = out_path / model_name / f"{model_name}.skops"
304
- with open(model_pickle_file, mode="bw") as f:
305
- dump(final_lda, file=f)
306
 
307
  # %%
308
  import os
@@ -311,7 +311,7 @@ from pathlib import Path
311
  from skops import card, hub_utils
312
 
313
  hub_out_path = Path(
314
- "troglobyte-sandbox/models/local_compartment_classifier_bd_boxes/hub"
315
  )
316
  if not hub_out_path.exists():
317
  hub_utils.init(
 
37
 
38
  # %%
39
 
40
+ X_df = pd.read_csv(out_path / model_name / "features_new.csv", index_col=[0, 1])
41
 
42
 
43
  # %%
 
300
  report_by_class.reset_index(inplace=True)
301
  # %%
302
 
303
+ # model_pickle_file = out_path / model_name / f"{model_name}.skops"
304
+ # with open(model_pickle_file, mode="bw") as f:
305
+ # dump(final_lda, file=f)
306
 
307
  # %%
308
  import os
 
311
  from skops import card, hub_utils
312
 
313
  hub_out_path = Path(
314
+ "troglobyte-sandbox/models/local_compartment_classifier_bd_boxes/hub_model"
315
  )
316
  if not hub_out_path.exists():
317
  hub_utils.init(