Spaces:
Sleeping
Sleeping
File size: 1,120 Bytes
e75a247 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
import os
from tqdm import tqdm
import argparse
import numpy as np
import pandas as pd
import pickle
from src.dataset.get_dataset import get_iter
from src.utils.paths import get_path
from pathlib import Path
# This script attempts to open dataset files and prints the number of events in each one.
parser = argparse.ArgumentParser()
parser.add_argument("--input", type=str, required=True)
parser.add_argument("--dataset-cap", type=int, default=-1)
parser.add_argument("--output", type=str, default="")
parser.add_argument("--plot-only", action="store_true")
args = parser.parse_args()
path = get_path(args.input, "preprocessed_data")
if args.output == "":
args.output = args.input
output_path = os.path.join(get_path(args.output, "results"), "analysis_name")
Path(output_path).mkdir(parents=True, exist_ok=True)
if not args.plot_only:
pass
# Do some computations here
# pickle.dump(result, open(os.path.join(output_path, "result.pkl"), "wb"))
#if args.plot_only:
# result = pickle.load(open(os.path.join(output_path, "result.pkl"), "rb"))
import matplotlib.pyplot as plt
# Do some plotting here
|