diff options
| -rw-r--r-- | .gitignore | 1 | ||||
| -rw-r--r-- | python-pelutils.spec | 1094 | ||||
| -rw-r--r-- | sources | 1 |
3 files changed, 1096 insertions, 0 deletions
@@ -0,0 +1 @@ +/pelutils-2.0.0.tar.gz diff --git a/python-pelutils.spec b/python-pelutils.spec new file mode 100644 index 0000000..76e9271 --- /dev/null +++ b/python-pelutils.spec @@ -0,0 +1,1094 @@ +%global _empty_manifest_terminate_build 0 +Name: python-pelutils +Version: 2.0.0 +Release: 1 +Summary: Utility functions that are often useful +License: BSD-3-Clause +URL: https://github.com/peleiden/pelutils +Source0: https://mirrors.nju.edu.cn/pypi/web/packages/fe/43/105c67a5b5c704d675e9cc9414c190daaeb449927bebc06e95c77d6800ab/pelutils-2.0.0.tar.gz + +Requires: python3-numpy +Requires: python3-gitpython +Requires: python3-rich +Requires: python3-click +Requires: python3-rapidjson +Requires: python3-py-cpuinfo +Requires: python3-psutil +Requires: python3-torch +Requires: python3-matplotlib +Requires: python3-scipy +Requires: python3-tqdm +Requires: python3-torch +Requires: python3-matplotlib +Requires: python3-scipy +Requires: python3-tqdm +Requires: python3-pytest +Requires: python3-pytest-cov +Requires: python3-coveralls +Requires: python3-coverage +Requires: python3-wheel +Requires: python3-setuptools + +%description +# pelutils + +Various utilities useful for Python projects. Features include + +- A simple and powerful logger with colourful printing and stacktrace logging +- Parsing for combining config files and command-line arguments - especially useful for algorithms with several parameters +- A timer inspired by Matlab's `tic` and `toc` +- Simple code profiler +- An extension to the built-in `dataclass` for saving and loading data +- Table formatting +- Miscellaneous standalone functions - see `pelutils/__init__.py` +- Data-science submodule with extra utilities for statistics, plotting with `matplotlib`, and machine learning using `PyTorch` +- Linear time `unique` function in the style of `numpy.unique` + +`pelutils` supports Python 3.7+. + +[](https://github.com/peleiden/pelutils/actions/workflows/pytest.yml) +[](https://coveralls.io/github/peleiden/pelutils?branch=master) + +## Timing and Code Profiling + +Simple time taker inspired by Matlab Tic, Toc, which also has profiling tooling. + +```py +# Time a task +TT.tick() +<some task> +seconds_used = TT.tock() + +# Profile a for loop +for i in range(100): + TT.profile("Repeated code") + <some task> + TT.profile("Subtask") + <some subtask> + TT.end_profile() + TT.end_profile() +print(TT) # Prints a table view of profiled code sections + +# Alternative syntax using with statement +with TT.profile("The best task"): + <some task> + +# When using multiprocessing, it can be useful to simulate multiple hits of the same profile +with mp.Pool() as p, TT.profile("Processing 100 items on multiple threads", hits=100): + p.map(100 items) +# Similar for very quick loops +a = 0 +with TT.profile("Adding 1 to a", hits=100): + for _ in range(100): + a += 1 + +# Examples so far use a global TickTock instance, which is convenient, +# but it can also be desirable to use for multiple different timers, e.g. +tt1 = TickTock() +tt2 = TickTock() +t1_interval = 1 # Do task 1 every second +t2_interval = 2 # Do task 2 every other second +tt1.tick() +tt2.tick() +while True: + if tt1.tock() > t1_interval: + <task 1> + tt1.tick() + if tt2.tock() > t2_interval: + <task 2> + tt2.tick() + time.sleep(0.01) +``` + +## Data Storage + +The DataStorage class is an augmentation of the dataclass that incluces save and load functionality. +This simplifies saving data, as only save command has to be issued for all data, and it keeps type hinting when loading data compared to e.g. a dictionary. + +Currently works specifically with: + +- Numpy arrays (`numpy.ndarray`) +- Torch tensors (`torch.Tensor`) +- Any `json` serializable data (as determined by the `rapidjson` library) + +All other data is pickled. + +DataStorage classes must inherit from DataStorage and be annotated with `@dataclass`. + +It is further possible to give arguments to the class definition: + +- `json_name`: Name of the saved json file +- `indent`: How many spaces to use for indenting in the json file + +Usage example: + +```py +@dataclass +class ResultData(DataStorage, json_name="game.json", indent=4): + shots: int + goalscorers: list + dists: np.ndarray + +rdata = ResultData(shots=1, goalscorers=["Max Fenger"], dists=np.ones(22)*10) +rdata.save("max") +# Now shots and goalscorers are saved in <pwd>/max/game.json and dists in <pwd>/max/dists.npy + +# Then to load +rdata = ResultData.load("max") +print(rdata.goalscorers) # ["Max Fenger"] +``` + +## Parsing + +A parsing tool for combining command-line and config file arguments. +Useful for parametric methods such as machine learning. +The first argument must always be a path. This can for instance be used to put log files, results, plots etc. + +Consider the execution of a file `main.py` with the command line call +``` +python main.py path/to/put/results -c path/to/config/file.ini --data-path path/to/data +``` +The config file could contain +``` +[DEFAULT] +fp16 +learning-rate=1e-4 + +[LOWLR] +learning-rate=1e-5 + +[NOFP16] +fp16=False +``` +where `main.py` contains +```py +options = [ + # Mandatory argument with set abbreviation -p + Argument("data-path", help="Path to where data is located", abbrv"-p"), + # Optional argument with auto-generated abbreviation -l + Option("learning-rate", default=1e-5, help="Learning rate to use for gradient descent steps"), + # Boolean flag with auto-generated abbreviation -f + Flag("fp16", help="Use mixed precision for training"), +] +parser = Parser(*options, multiple_jobs=True) # Two jobs are specified in the config file, so multiple_jobs=True +location = parser.location # Experiments are stored here. In this case path/to/put/results +job_descriptions = parser.parse_args() +# Run each experiment +for job in job_descriptions: + # Clear directory where job is located and put a documentation file there + job.prepare_directory() + # Get location of this job as job.location + run_experiment(job) +``` + +This could then by run by +`python main.py data/my-big-experiment --learning-rate 1e-5` +or by +`python main.py data/my-big-experiment --config cfg.ini` +or using a combination where CLI args takes precedence: +`python main.py data/my-big-experiment --config cfg.ini --learning-rate 1e-5` +where `cfg.ini` could contain + +# Logging + +The logging submodule contains a simple yet feature-rich logger which fits common needs. Can be imported from `pelutils` directly, e.g. `from pelutils import log`. + +```py +# Configure logger for the script +log.configure("path/to/save/log.log") + +# Start logging +for i in range(70): # Nice + log("Execution %i" % i) + +# Sections +log.section("New section in the logfile") + +# Adjust logging levels +log.warning("Will be logged") +with log.level(LogLevels.ERROR): # Only log at ERROR level or above + log.warning("Will not be logged") +with log.no_log: + log.section("I will not be logged") + +# Error handling +# The zero-division error and stacktrace is logged +with log.log_errors: + 0 / 0 +# Entire chained stacktrace is logged +with log.log_errors: + try: + 0 / 0 + except ZeroDivisionError as e: + raise ValueError("Denominator must be non-zero") from e + +# User input - acts like built-in input but logs both prompt and user input +inp = log.input("Continue [Y/n]? ") +# Parse yes/no user input +cont = log.parse_bool_input(inp, default=True) + +# Log all logs from a function at the same time +# This is especially useful when using multiple threads so logging does not get mixed up +def fun(): + log("Hello there") + log("General Kenobi!") +with mp.Pool() as p: + p.map(log.collect_logs(fun), args) + +# It is also possible to create multiple loggers by importing the Logger class, e.g. +log2 = Logger() +log2.configure("path/to/save/log2.log") +``` + +# Data Science + +This submodule contains various utility functions for data science and machine learning. To make sure the necessary requirements are installed, install using +``` +pip install pelutils[ds] +``` +Note that in some terminals (e.g. `zsh`), you will have escape the brackets: +``` +pip install pelutils\[ds\] +``` + +## Deep Learning + +All PyTorch functions work independently of whether CUDA is available or not. + +```py +# Inference only: No gradients should be tracked in the following function +# Same as putting entire function body inside `with torch.no_grad()` +@no_grad +def infer(): + <code that includes feedforwarding> +``` + +## Statistics + +Includes various commonly used statistical functions. + +```py +# Get one sided z value for exponential(lambda=2) distribution with a significance level of 1 % +zval = z(alpha=0.01, two_sided=False, distribution=scipy.stats.expon(loc=1/2)) + +# Get correlation, confidence interval, and p value for two vectors +a, b = np.random.randn(100), np.random.randn(100) +r, lower_r, upper_r, p = corr_ci(a, b, alpha=0.01) +``` + +## Plotting + +`pelutils` provides plotting utilities based on `matplotlib`. +Most notable is the `Figure` context class, which attempts to remedy some of the common grievances with `matplotlib`, e.g. having to remember the correct `kwargs` and `rcParams` for setting font sizes, grid line colours etc, and notably adding type hinting to `fig` and `ax` produced by `plt.subplots`. +```py +from pelutils.ds.plots import Figure + +# The following makes a plot and saves it to `plot.png`. +# The seaborn is style is used for demonstration, but if the `style` argument +# is not given, the default matplotlib style is used. +# The figure and font size are also given for demonstration, but their default +# values are increased compared to matplotlib's default, as these are generally +# too small for finished plots. +with Figure("plot.png", figsize=(20, 10), style="seaborn", fontsize=20): + plt.scatter(x, y, label="Data") + plt.grid() + plt.title("Very nice plot") +# The figure is automatically saved to `plot.png` and closed, such that +# plt.plot can be used again from here. +# Figure changes `matplotlib.rcParams`, but these changes are also undone +# after the end of the `with statement`. + +# For more complex plots, it is also possible to access the `fig` and `ax` +# variables usually assigned as `fig, ax = plt.subplots()`. +# These are type hinted, so no more remembering if it is `ax.title()` or +# `ax.set_title()`. +with Figure("plot.png") as f: + f.fig # fig available as attribute on the Figure instance + f.ax.set_title("Very nice plot") # The same goes for `ax` +``` + +The plotting utilies also include binning functions for creating nice histograms. +The `get_bins` function produces bins based on a binning function, of which three are provided: + +- `linear_binning`: Bins are spaced evenly from the lowest to the largest value of the data. +- `log_binning`: Bins are log-spaced from the lowest to the largest value of the data, which is assumed to be positive. +- `normal_binning`: Bins are distributed according to the distribution of the data, such there are more bins closer to the center of the data. This is useful if the data somewhat resembles a normal distribution, as the resolution will be the greatest where there is the most data. + +It is also possible to provide custom binning functions. + +`get_bins` provide both `x` and `y` coordinates, making it simple to use with argument unpacking: +```py +import matplotlib.pyplot as plt +import numpy as np +from pelutils.ds.plots import get_bins, normal_binning + +# Generate normally distributed data +x = np.random.randn(100) +# Plot distribution +plt.plot(*get_bins(x, binning_fn=normal_binning)) +``` + +Finally, different smoothing functions are provided. +The two most common are `moving_avg` and `exponential_avg` which smooth the data using a moving average and exponential smoothing, respectively. + +The `double_moving_avg` is special in that the number of smoothed data points do not depend on the number of given data points but is instead based on a given number of samples, which allows the resulting smoothed curve to not by jagged as happens with the other smoothing functions. +It also has two smoothness parameters, which allows a large degree of smoothness control. + +Apart from smoothness parameters, all smoothness functions have the same call signature: +```py +from pelutils.ds.plots import double_moving_avg + +# Generate noisy data +n = 100 +x = np.linspace(-1, 1, n) +y = np.random.randn(n) + +# Plot data along with smoothed curve +plt.plot(*double_moving_avg(x, y)) +# If x is not given, it is assumed to go from 0 to n-1 in steps of 1 +plt.plot(*double_moving_avg(y)) +``` + +Examples of all the plotting utilities are shown in the `examples` directory. + +# Supported platforms + +Precompiled wheels are provided for most common platforms. +Notably, they are not provided for 32-bit systems. +If no wheel is provided, `pip` should attempt a source install. +If all else fails, it is possible to install from source by pointing `pip` to Github directly: +``` +pip install git+https://github.com/peleiden/pelutils.git@release#egg=pelutils +``` +It is also possible to install from source using `pip`'s `--no-binary` option. + + + + +%package -n python3-pelutils +Summary: Utility functions that are often useful +Provides: python-pelutils +BuildRequires: python3-devel +BuildRequires: python3-setuptools +BuildRequires: python3-pip +BuildRequires: python3-cffi +BuildRequires: gcc +BuildRequires: gdb +%description -n python3-pelutils +# pelutils + +Various utilities useful for Python projects. Features include + +- A simple and powerful logger with colourful printing and stacktrace logging +- Parsing for combining config files and command-line arguments - especially useful for algorithms with several parameters +- A timer inspired by Matlab's `tic` and `toc` +- Simple code profiler +- An extension to the built-in `dataclass` for saving and loading data +- Table formatting +- Miscellaneous standalone functions - see `pelutils/__init__.py` +- Data-science submodule with extra utilities for statistics, plotting with `matplotlib`, and machine learning using `PyTorch` +- Linear time `unique` function in the style of `numpy.unique` + +`pelutils` supports Python 3.7+. + +[](https://github.com/peleiden/pelutils/actions/workflows/pytest.yml) +[](https://coveralls.io/github/peleiden/pelutils?branch=master) + +## Timing and Code Profiling + +Simple time taker inspired by Matlab Tic, Toc, which also has profiling tooling. + +```py +# Time a task +TT.tick() +<some task> +seconds_used = TT.tock() + +# Profile a for loop +for i in range(100): + TT.profile("Repeated code") + <some task> + TT.profile("Subtask") + <some subtask> + TT.end_profile() + TT.end_profile() +print(TT) # Prints a table view of profiled code sections + +# Alternative syntax using with statement +with TT.profile("The best task"): + <some task> + +# When using multiprocessing, it can be useful to simulate multiple hits of the same profile +with mp.Pool() as p, TT.profile("Processing 100 items on multiple threads", hits=100): + p.map(100 items) +# Similar for very quick loops +a = 0 +with TT.profile("Adding 1 to a", hits=100): + for _ in range(100): + a += 1 + +# Examples so far use a global TickTock instance, which is convenient, +# but it can also be desirable to use for multiple different timers, e.g. +tt1 = TickTock() +tt2 = TickTock() +t1_interval = 1 # Do task 1 every second +t2_interval = 2 # Do task 2 every other second +tt1.tick() +tt2.tick() +while True: + if tt1.tock() > t1_interval: + <task 1> + tt1.tick() + if tt2.tock() > t2_interval: + <task 2> + tt2.tick() + time.sleep(0.01) +``` + +## Data Storage + +The DataStorage class is an augmentation of the dataclass that incluces save and load functionality. +This simplifies saving data, as only save command has to be issued for all data, and it keeps type hinting when loading data compared to e.g. a dictionary. + +Currently works specifically with: + +- Numpy arrays (`numpy.ndarray`) +- Torch tensors (`torch.Tensor`) +- Any `json` serializable data (as determined by the `rapidjson` library) + +All other data is pickled. + +DataStorage classes must inherit from DataStorage and be annotated with `@dataclass`. + +It is further possible to give arguments to the class definition: + +- `json_name`: Name of the saved json file +- `indent`: How many spaces to use for indenting in the json file + +Usage example: + +```py +@dataclass +class ResultData(DataStorage, json_name="game.json", indent=4): + shots: int + goalscorers: list + dists: np.ndarray + +rdata = ResultData(shots=1, goalscorers=["Max Fenger"], dists=np.ones(22)*10) +rdata.save("max") +# Now shots and goalscorers are saved in <pwd>/max/game.json and dists in <pwd>/max/dists.npy + +# Then to load +rdata = ResultData.load("max") +print(rdata.goalscorers) # ["Max Fenger"] +``` + +## Parsing + +A parsing tool for combining command-line and config file arguments. +Useful for parametric methods such as machine learning. +The first argument must always be a path. This can for instance be used to put log files, results, plots etc. + +Consider the execution of a file `main.py` with the command line call +``` +python main.py path/to/put/results -c path/to/config/file.ini --data-path path/to/data +``` +The config file could contain +``` +[DEFAULT] +fp16 +learning-rate=1e-4 + +[LOWLR] +learning-rate=1e-5 + +[NOFP16] +fp16=False +``` +where `main.py` contains +```py +options = [ + # Mandatory argument with set abbreviation -p + Argument("data-path", help="Path to where data is located", abbrv"-p"), + # Optional argument with auto-generated abbreviation -l + Option("learning-rate", default=1e-5, help="Learning rate to use for gradient descent steps"), + # Boolean flag with auto-generated abbreviation -f + Flag("fp16", help="Use mixed precision for training"), +] +parser = Parser(*options, multiple_jobs=True) # Two jobs are specified in the config file, so multiple_jobs=True +location = parser.location # Experiments are stored here. In this case path/to/put/results +job_descriptions = parser.parse_args() +# Run each experiment +for job in job_descriptions: + # Clear directory where job is located and put a documentation file there + job.prepare_directory() + # Get location of this job as job.location + run_experiment(job) +``` + +This could then by run by +`python main.py data/my-big-experiment --learning-rate 1e-5` +or by +`python main.py data/my-big-experiment --config cfg.ini` +or using a combination where CLI args takes precedence: +`python main.py data/my-big-experiment --config cfg.ini --learning-rate 1e-5` +where `cfg.ini` could contain + +# Logging + +The logging submodule contains a simple yet feature-rich logger which fits common needs. Can be imported from `pelutils` directly, e.g. `from pelutils import log`. + +```py +# Configure logger for the script +log.configure("path/to/save/log.log") + +# Start logging +for i in range(70): # Nice + log("Execution %i" % i) + +# Sections +log.section("New section in the logfile") + +# Adjust logging levels +log.warning("Will be logged") +with log.level(LogLevels.ERROR): # Only log at ERROR level or above + log.warning("Will not be logged") +with log.no_log: + log.section("I will not be logged") + +# Error handling +# The zero-division error and stacktrace is logged +with log.log_errors: + 0 / 0 +# Entire chained stacktrace is logged +with log.log_errors: + try: + 0 / 0 + except ZeroDivisionError as e: + raise ValueError("Denominator must be non-zero") from e + +# User input - acts like built-in input but logs both prompt and user input +inp = log.input("Continue [Y/n]? ") +# Parse yes/no user input +cont = log.parse_bool_input(inp, default=True) + +# Log all logs from a function at the same time +# This is especially useful when using multiple threads so logging does not get mixed up +def fun(): + log("Hello there") + log("General Kenobi!") +with mp.Pool() as p: + p.map(log.collect_logs(fun), args) + +# It is also possible to create multiple loggers by importing the Logger class, e.g. +log2 = Logger() +log2.configure("path/to/save/log2.log") +``` + +# Data Science + +This submodule contains various utility functions for data science and machine learning. To make sure the necessary requirements are installed, install using +``` +pip install pelutils[ds] +``` +Note that in some terminals (e.g. `zsh`), you will have escape the brackets: +``` +pip install pelutils\[ds\] +``` + +## Deep Learning + +All PyTorch functions work independently of whether CUDA is available or not. + +```py +# Inference only: No gradients should be tracked in the following function +# Same as putting entire function body inside `with torch.no_grad()` +@no_grad +def infer(): + <code that includes feedforwarding> +``` + +## Statistics + +Includes various commonly used statistical functions. + +```py +# Get one sided z value for exponential(lambda=2) distribution with a significance level of 1 % +zval = z(alpha=0.01, two_sided=False, distribution=scipy.stats.expon(loc=1/2)) + +# Get correlation, confidence interval, and p value for two vectors +a, b = np.random.randn(100), np.random.randn(100) +r, lower_r, upper_r, p = corr_ci(a, b, alpha=0.01) +``` + +## Plotting + +`pelutils` provides plotting utilities based on `matplotlib`. +Most notable is the `Figure` context class, which attempts to remedy some of the common grievances with `matplotlib`, e.g. having to remember the correct `kwargs` and `rcParams` for setting font sizes, grid line colours etc, and notably adding type hinting to `fig` and `ax` produced by `plt.subplots`. +```py +from pelutils.ds.plots import Figure + +# The following makes a plot and saves it to `plot.png`. +# The seaborn is style is used for demonstration, but if the `style` argument +# is not given, the default matplotlib style is used. +# The figure and font size are also given for demonstration, but their default +# values are increased compared to matplotlib's default, as these are generally +# too small for finished plots. +with Figure("plot.png", figsize=(20, 10), style="seaborn", fontsize=20): + plt.scatter(x, y, label="Data") + plt.grid() + plt.title("Very nice plot") +# The figure is automatically saved to `plot.png` and closed, such that +# plt.plot can be used again from here. +# Figure changes `matplotlib.rcParams`, but these changes are also undone +# after the end of the `with statement`. + +# For more complex plots, it is also possible to access the `fig` and `ax` +# variables usually assigned as `fig, ax = plt.subplots()`. +# These are type hinted, so no more remembering if it is `ax.title()` or +# `ax.set_title()`. +with Figure("plot.png") as f: + f.fig # fig available as attribute on the Figure instance + f.ax.set_title("Very nice plot") # The same goes for `ax` +``` + +The plotting utilies also include binning functions for creating nice histograms. +The `get_bins` function produces bins based on a binning function, of which three are provided: + +- `linear_binning`: Bins are spaced evenly from the lowest to the largest value of the data. +- `log_binning`: Bins are log-spaced from the lowest to the largest value of the data, which is assumed to be positive. +- `normal_binning`: Bins are distributed according to the distribution of the data, such there are more bins closer to the center of the data. This is useful if the data somewhat resembles a normal distribution, as the resolution will be the greatest where there is the most data. + +It is also possible to provide custom binning functions. + +`get_bins` provide both `x` and `y` coordinates, making it simple to use with argument unpacking: +```py +import matplotlib.pyplot as plt +import numpy as np +from pelutils.ds.plots import get_bins, normal_binning + +# Generate normally distributed data +x = np.random.randn(100) +# Plot distribution +plt.plot(*get_bins(x, binning_fn=normal_binning)) +``` + +Finally, different smoothing functions are provided. +The two most common are `moving_avg` and `exponential_avg` which smooth the data using a moving average and exponential smoothing, respectively. + +The `double_moving_avg` is special in that the number of smoothed data points do not depend on the number of given data points but is instead based on a given number of samples, which allows the resulting smoothed curve to not by jagged as happens with the other smoothing functions. +It also has two smoothness parameters, which allows a large degree of smoothness control. + +Apart from smoothness parameters, all smoothness functions have the same call signature: +```py +from pelutils.ds.plots import double_moving_avg + +# Generate noisy data +n = 100 +x = np.linspace(-1, 1, n) +y = np.random.randn(n) + +# Plot data along with smoothed curve +plt.plot(*double_moving_avg(x, y)) +# If x is not given, it is assumed to go from 0 to n-1 in steps of 1 +plt.plot(*double_moving_avg(y)) +``` + +Examples of all the plotting utilities are shown in the `examples` directory. + +# Supported platforms + +Precompiled wheels are provided for most common platforms. +Notably, they are not provided for 32-bit systems. +If no wheel is provided, `pip` should attempt a source install. +If all else fails, it is possible to install from source by pointing `pip` to Github directly: +``` +pip install git+https://github.com/peleiden/pelutils.git@release#egg=pelutils +``` +It is also possible to install from source using `pip`'s `--no-binary` option. + + + + +%package help +Summary: Development documents and examples for pelutils +Provides: python3-pelutils-doc +%description help +# pelutils + +Various utilities useful for Python projects. Features include + +- A simple and powerful logger with colourful printing and stacktrace logging +- Parsing for combining config files and command-line arguments - especially useful for algorithms with several parameters +- A timer inspired by Matlab's `tic` and `toc` +- Simple code profiler +- An extension to the built-in `dataclass` for saving and loading data +- Table formatting +- Miscellaneous standalone functions - see `pelutils/__init__.py` +- Data-science submodule with extra utilities for statistics, plotting with `matplotlib`, and machine learning using `PyTorch` +- Linear time `unique` function in the style of `numpy.unique` + +`pelutils` supports Python 3.7+. + +[](https://github.com/peleiden/pelutils/actions/workflows/pytest.yml) +[](https://coveralls.io/github/peleiden/pelutils?branch=master) + +## Timing and Code Profiling + +Simple time taker inspired by Matlab Tic, Toc, which also has profiling tooling. + +```py +# Time a task +TT.tick() +<some task> +seconds_used = TT.tock() + +# Profile a for loop +for i in range(100): + TT.profile("Repeated code") + <some task> + TT.profile("Subtask") + <some subtask> + TT.end_profile() + TT.end_profile() +print(TT) # Prints a table view of profiled code sections + +# Alternative syntax using with statement +with TT.profile("The best task"): + <some task> + +# When using multiprocessing, it can be useful to simulate multiple hits of the same profile +with mp.Pool() as p, TT.profile("Processing 100 items on multiple threads", hits=100): + p.map(100 items) +# Similar for very quick loops +a = 0 +with TT.profile("Adding 1 to a", hits=100): + for _ in range(100): + a += 1 + +# Examples so far use a global TickTock instance, which is convenient, +# but it can also be desirable to use for multiple different timers, e.g. +tt1 = TickTock() +tt2 = TickTock() +t1_interval = 1 # Do task 1 every second +t2_interval = 2 # Do task 2 every other second +tt1.tick() +tt2.tick() +while True: + if tt1.tock() > t1_interval: + <task 1> + tt1.tick() + if tt2.tock() > t2_interval: + <task 2> + tt2.tick() + time.sleep(0.01) +``` + +## Data Storage + +The DataStorage class is an augmentation of the dataclass that incluces save and load functionality. +This simplifies saving data, as only save command has to be issued for all data, and it keeps type hinting when loading data compared to e.g. a dictionary. + +Currently works specifically with: + +- Numpy arrays (`numpy.ndarray`) +- Torch tensors (`torch.Tensor`) +- Any `json` serializable data (as determined by the `rapidjson` library) + +All other data is pickled. + +DataStorage classes must inherit from DataStorage and be annotated with `@dataclass`. + +It is further possible to give arguments to the class definition: + +- `json_name`: Name of the saved json file +- `indent`: How many spaces to use for indenting in the json file + +Usage example: + +```py +@dataclass +class ResultData(DataStorage, json_name="game.json", indent=4): + shots: int + goalscorers: list + dists: np.ndarray + +rdata = ResultData(shots=1, goalscorers=["Max Fenger"], dists=np.ones(22)*10) +rdata.save("max") +# Now shots and goalscorers are saved in <pwd>/max/game.json and dists in <pwd>/max/dists.npy + +# Then to load +rdata = ResultData.load("max") +print(rdata.goalscorers) # ["Max Fenger"] +``` + +## Parsing + +A parsing tool for combining command-line and config file arguments. +Useful for parametric methods such as machine learning. +The first argument must always be a path. This can for instance be used to put log files, results, plots etc. + +Consider the execution of a file `main.py` with the command line call +``` +python main.py path/to/put/results -c path/to/config/file.ini --data-path path/to/data +``` +The config file could contain +``` +[DEFAULT] +fp16 +learning-rate=1e-4 + +[LOWLR] +learning-rate=1e-5 + +[NOFP16] +fp16=False +``` +where `main.py` contains +```py +options = [ + # Mandatory argument with set abbreviation -p + Argument("data-path", help="Path to where data is located", abbrv"-p"), + # Optional argument with auto-generated abbreviation -l + Option("learning-rate", default=1e-5, help="Learning rate to use for gradient descent steps"), + # Boolean flag with auto-generated abbreviation -f + Flag("fp16", help="Use mixed precision for training"), +] +parser = Parser(*options, multiple_jobs=True) # Two jobs are specified in the config file, so multiple_jobs=True +location = parser.location # Experiments are stored here. In this case path/to/put/results +job_descriptions = parser.parse_args() +# Run each experiment +for job in job_descriptions: + # Clear directory where job is located and put a documentation file there + job.prepare_directory() + # Get location of this job as job.location + run_experiment(job) +``` + +This could then by run by +`python main.py data/my-big-experiment --learning-rate 1e-5` +or by +`python main.py data/my-big-experiment --config cfg.ini` +or using a combination where CLI args takes precedence: +`python main.py data/my-big-experiment --config cfg.ini --learning-rate 1e-5` +where `cfg.ini` could contain + +# Logging + +The logging submodule contains a simple yet feature-rich logger which fits common needs. Can be imported from `pelutils` directly, e.g. `from pelutils import log`. + +```py +# Configure logger for the script +log.configure("path/to/save/log.log") + +# Start logging +for i in range(70): # Nice + log("Execution %i" % i) + +# Sections +log.section("New section in the logfile") + +# Adjust logging levels +log.warning("Will be logged") +with log.level(LogLevels.ERROR): # Only log at ERROR level or above + log.warning("Will not be logged") +with log.no_log: + log.section("I will not be logged") + +# Error handling +# The zero-division error and stacktrace is logged +with log.log_errors: + 0 / 0 +# Entire chained stacktrace is logged +with log.log_errors: + try: + 0 / 0 + except ZeroDivisionError as e: + raise ValueError("Denominator must be non-zero") from e + +# User input - acts like built-in input but logs both prompt and user input +inp = log.input("Continue [Y/n]? ") +# Parse yes/no user input +cont = log.parse_bool_input(inp, default=True) + +# Log all logs from a function at the same time +# This is especially useful when using multiple threads so logging does not get mixed up +def fun(): + log("Hello there") + log("General Kenobi!") +with mp.Pool() as p: + p.map(log.collect_logs(fun), args) + +# It is also possible to create multiple loggers by importing the Logger class, e.g. +log2 = Logger() +log2.configure("path/to/save/log2.log") +``` + +# Data Science + +This submodule contains various utility functions for data science and machine learning. To make sure the necessary requirements are installed, install using +``` +pip install pelutils[ds] +``` +Note that in some terminals (e.g. `zsh`), you will have escape the brackets: +``` +pip install pelutils\[ds\] +``` + +## Deep Learning + +All PyTorch functions work independently of whether CUDA is available or not. + +```py +# Inference only: No gradients should be tracked in the following function +# Same as putting entire function body inside `with torch.no_grad()` +@no_grad +def infer(): + <code that includes feedforwarding> +``` + +## Statistics + +Includes various commonly used statistical functions. + +```py +# Get one sided z value for exponential(lambda=2) distribution with a significance level of 1 % +zval = z(alpha=0.01, two_sided=False, distribution=scipy.stats.expon(loc=1/2)) + +# Get correlation, confidence interval, and p value for two vectors +a, b = np.random.randn(100), np.random.randn(100) +r, lower_r, upper_r, p = corr_ci(a, b, alpha=0.01) +``` + +## Plotting + +`pelutils` provides plotting utilities based on `matplotlib`. +Most notable is the `Figure` context class, which attempts to remedy some of the common grievances with `matplotlib`, e.g. having to remember the correct `kwargs` and `rcParams` for setting font sizes, grid line colours etc, and notably adding type hinting to `fig` and `ax` produced by `plt.subplots`. +```py +from pelutils.ds.plots import Figure + +# The following makes a plot and saves it to `plot.png`. +# The seaborn is style is used for demonstration, but if the `style` argument +# is not given, the default matplotlib style is used. +# The figure and font size are also given for demonstration, but their default +# values are increased compared to matplotlib's default, as these are generally +# too small for finished plots. +with Figure("plot.png", figsize=(20, 10), style="seaborn", fontsize=20): + plt.scatter(x, y, label="Data") + plt.grid() + plt.title("Very nice plot") +# The figure is automatically saved to `plot.png` and closed, such that +# plt.plot can be used again from here. +# Figure changes `matplotlib.rcParams`, but these changes are also undone +# after the end of the `with statement`. + +# For more complex plots, it is also possible to access the `fig` and `ax` +# variables usually assigned as `fig, ax = plt.subplots()`. +# These are type hinted, so no more remembering if it is `ax.title()` or +# `ax.set_title()`. +with Figure("plot.png") as f: + f.fig # fig available as attribute on the Figure instance + f.ax.set_title("Very nice plot") # The same goes for `ax` +``` + +The plotting utilies also include binning functions for creating nice histograms. +The `get_bins` function produces bins based on a binning function, of which three are provided: + +- `linear_binning`: Bins are spaced evenly from the lowest to the largest value of the data. +- `log_binning`: Bins are log-spaced from the lowest to the largest value of the data, which is assumed to be positive. +- `normal_binning`: Bins are distributed according to the distribution of the data, such there are more bins closer to the center of the data. This is useful if the data somewhat resembles a normal distribution, as the resolution will be the greatest where there is the most data. + +It is also possible to provide custom binning functions. + +`get_bins` provide both `x` and `y` coordinates, making it simple to use with argument unpacking: +```py +import matplotlib.pyplot as plt +import numpy as np +from pelutils.ds.plots import get_bins, normal_binning + +# Generate normally distributed data +x = np.random.randn(100) +# Plot distribution +plt.plot(*get_bins(x, binning_fn=normal_binning)) +``` + +Finally, different smoothing functions are provided. +The two most common are `moving_avg` and `exponential_avg` which smooth the data using a moving average and exponential smoothing, respectively. + +The `double_moving_avg` is special in that the number of smoothed data points do not depend on the number of given data points but is instead based on a given number of samples, which allows the resulting smoothed curve to not by jagged as happens with the other smoothing functions. +It also has two smoothness parameters, which allows a large degree of smoothness control. + +Apart from smoothness parameters, all smoothness functions have the same call signature: +```py +from pelutils.ds.plots import double_moving_avg + +# Generate noisy data +n = 100 +x = np.linspace(-1, 1, n) +y = np.random.randn(n) + +# Plot data along with smoothed curve +plt.plot(*double_moving_avg(x, y)) +# If x is not given, it is assumed to go from 0 to n-1 in steps of 1 +plt.plot(*double_moving_avg(y)) +``` + +Examples of all the plotting utilities are shown in the `examples` directory. + +# Supported platforms + +Precompiled wheels are provided for most common platforms. +Notably, they are not provided for 32-bit systems. +If no wheel is provided, `pip` should attempt a source install. +If all else fails, it is possible to install from source by pointing `pip` to Github directly: +``` +pip install git+https://github.com/peleiden/pelutils.git@release#egg=pelutils +``` +It is also possible to install from source using `pip`'s `--no-binary` option. + + + + +%prep +%autosetup -n pelutils-2.0.0 + +%build +%py3_build + +%install +%py3_install +install -d -m755 %{buildroot}/%{_pkgdocdir} +if [ -d doc ]; then cp -arf doc %{buildroot}/%{_pkgdocdir}; fi +if [ -d docs ]; then cp -arf docs %{buildroot}/%{_pkgdocdir}; fi +if [ -d example ]; then cp -arf example %{buildroot}/%{_pkgdocdir}; fi +if [ -d examples ]; then cp -arf examples %{buildroot}/%{_pkgdocdir}; fi +pushd %{buildroot} +if [ -d usr/lib ]; then + find usr/lib -type f -printf "/%h/%f\n" >> filelist.lst +fi +if [ -d usr/lib64 ]; then + find usr/lib64 -type f -printf "/%h/%f\n" >> filelist.lst +fi +if [ -d usr/bin ]; then + find usr/bin -type f -printf "/%h/%f\n" >> filelist.lst +fi +if [ -d usr/sbin ]; then + find usr/sbin -type f -printf "/%h/%f\n" >> filelist.lst +fi +touch doclist.lst +if [ -d usr/share/man ]; then + find usr/share/man -type f -printf "/%h/%f.gz\n" >> doclist.lst +fi +popd +mv %{buildroot}/filelist.lst . +mv %{buildroot}/doclist.lst . + +%files -n python3-pelutils -f filelist.lst +%dir %{python3_sitearch}/* + +%files help -f doclist.lst +%{_docdir}/* + +%changelog +* Mon May 29 2023 Python_Bot <Python_Bot@openeuler.org> - 2.0.0-1 +- Package Spec generated @@ -0,0 +1 @@ +a9f16560c89a0bafa221f630a8b65c63 pelutils-2.0.0.tar.gz |