Skip to content

Commit

Permalink
Update step2_dlq.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Kirandkv authored Nov 26, 2024
1 parent e3e5e21 commit 3c88deb
Showing 1 changed file with 10 additions and 11 deletions.
21 changes: 10 additions & 11 deletions code/step2_dlq.py
Original file line number Diff line number Diff line change
Expand Up @@ -398,24 +398,23 @@ def safe_float(series):
plt.close()

def main():
# Unzip input data files
input_zip_path = "./code/Input_data_step2.zip" # Adjust this path to your zip file
extract_path = "./code/input_data_step2"
input_zip_path = "./code/Input_data_step2.zip" # Adjust this path to your zip file
extract_path = "./code/input_data_step2"

# Create extraction directory if it doesn't exist
os.makedirs(extract_path, exist_ok=True)
# Create extraction directory if it doesn't exist
os.makedirs(extract_path, exist_ok=True)

# Unzip the input files
with zipfile.ZipFile(input_zip_path, 'r') as zip_ref:
zip_ref.extractall(extract_path)
print(f"Input files extracted to: {extract_path}")
# Unzip the input files
with zipfile.ZipFile(input_zip_path, 'r') as zip_ref:
zip_ref.extractall(extract_path)
print(f"Input files extracted to: {extract_path}")

# Configuration
config = {
'gap_time': 5,
'length_threshold': 2,
'simulation_data_path': "./input_data_step2/simulation_output_new.pkl",
'output_file_path': "path to save output",
'simulation_data_path': "./code/input_data_step2/simulation_output_new.pkl",
'output_file_path': 'path to output_data folder/',
}

# Create output directory
Expand Down

0 comments on commit 3c88deb

Please sign in to comment.