A low-code library for machine learning pipelines
Project description
blitzml
Automate machine learning piplines rapidly
How to install
pip install blitzml
Usage
from blitzml.csv import Pipeline
dataset_folder = "auxiliary/data/" # contains train.csv and test.csv
ground_truth_path = "auxiliary/ground_truth.csv"
output_folder_path = "auxiliary/output/"
auto = Pipeline(dataset_folder, ground_truth_path, output_folder_path, classifier = 'RF', n_estimators = 50)
auto.preprocess()
auto.train_the_model()
auto.gen_metrics_dict()
metrics_dict = auto.metrics_dict
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
blitzml-0.2.0.tar.gz
(4.2 kB
view hashes)