Ravix

Ravix is a Python package for regression analysis and data visualization. It provides tools for model fitting, prediction, and various types of plots to help visualize your data.

Features

• Model fitting and prediction with convenient formula notation
• Streamlined code for plotting (boxplot, histogram, scatter plot, etc.)
• Regression analysis diagnostic tools
• Integration with popular libraries like pandas and statsmodels

Installation

You can install the package using pip:

pip install ravix

Usage

Here are some examples of how to use the key functions in the package.

Importing the Package

To use the functions provided by the package, import it as follows:

import ravix

Getting Data

There are multiple datasets available from Ravix and are easily attained using the get_data function. The datasets currently available can be found using the following:

import ravix
ravix.get_data()

See Applied Linear Regression for Business Analytics with Python for details regarding these datasets. Sample import example:

import ravix

# Load data from ravix
df = ravix.get_data("betas.csv")

# Format the data (for later)
df.drop(columns = df.columns[0], inplace=True)

Model Fitting and Prediction

Ravix formula supports formula functionality similar to R. Fit a model with a formula:

# Fit model with formula 
model = ravix.ols("SPY ~ .", df)

Summary types are specified using the out argument. Different summaries are available including:

• simple (default)
• statsmodels
• R
• ANOVA
• coefficients (coef)

# Generate a model summary
model.summary()

Making Predictions

A Statsmodels object is created by default. From this object, the predict function can be used. Since df is the dataframe used to fit the model, the following lines produce the same result.

# Make predictions
ravix.predict(model, df)

# Produce fitted values
ravix.predict(model)

General Plotting

Plotting code is streamlined and built on top of Seaborn and MatPlotLib. Samples provided below.

# Generate a boxplot
ravix.boxplot("SPY ~ .", df)

# Generate a histogram
ravix.hist(df.SPY)

# Multiple histograms
ravix.hist(df)

# Scatter plot
ravix.plot("MSFT ~ SPY", data = df)

# Multiple Scatter plots
ravix.plot(df)

# Correlation Plot
ravix.plot_cor(df)

Contributing

We welcome contributions to Ravix! If you find a bug or have a feature request, please open an issue on GitHub. You can also contribute by:

1. Forking the repository
2. Creating a new branch (git checkout -b feature-branch)
3. Committing your changes (git commit -am 'Add some feature')
4. Pushing to the branch (git push origin feature-branch)
5. Creating a pull request

License

This project is licensed under the MIT License - see the LICENSE file for details.

Acknowledgements

We would like to thank all contributors and users of Ravix for their support and feedback.