nirajandhakal commited on
Commit
4c6b419
·
verified ·
1 Parent(s): 3d63c38

Update evaluation_script.py

Browse files
Files changed (1) hide show
  1. evaluation_script.py +2 -2
evaluation_script.py CHANGED
@@ -38,7 +38,7 @@ policy, value = policy_value_net(dummy_input)
38
 
39
  # Load the weights (replace 'your_model.weights.h5' with your actual file)
40
  try:
41
- model_path = "/stockzero/models_weights/StockZero-2025-03-24-1727.weights.h5"
42
  policy_value_net.load_weights(model_path)
43
  print(f"Model weights loaded successfully from '{model_path}'")
44
  except Exception as e:
@@ -187,7 +187,7 @@ print("To convert to TensorFlow.js format, run the 'tensorflowjs_converter' comm
187
  try:
188
  current_datetime = datetime.datetime.now()
189
  zip_file_name = f"converted_models-{current_datetime.strftime('%Y%m%d%H%M')}"
190
- zip_file_path = f"/directory/{zip_file_name}"
191
  shutil.make_archive(zip_file_path, 'zip', OUTPUT_DIR) # Create zip archive
192
  print(f"All converted model files zipped to '{zip_file_path}.zip'")
193
  files.download(f"{zip_file_path}.zip") # Trigger download in Colab
 
38
 
39
  # Load the weights (replace 'your_model.weights.h5' with your actual file)
40
  try:
41
+ model_path = "/content/models_colab/StockZero-2025-03-24-1727.weights.h5"
42
  policy_value_net.load_weights(model_path)
43
  print(f"Model weights loaded successfully from '{model_path}'")
44
  except Exception as e:
 
187
  try:
188
  current_datetime = datetime.datetime.now()
189
  zip_file_name = f"converted_models-{current_datetime.strftime('%Y%m%d%H%M')}"
190
+ zip_file_path = f"/content/{zip_file_name}"
191
  shutil.make_archive(zip_file_path, 'zip', OUTPUT_DIR) # Create zip archive
192
  print(f"All converted model files zipped to '{zip_file_path}.zip'")
193
  files.download(f"{zip_file_path}.zip") # Trigger download in Colab