Skip to main content

A simple package to perform classification on a given dataset in csv or excel format

Project description

Machine Learning Classification Python Package

Functionality of the Package

  1. Performs Classification using the following algorithms with the default parameters:

    • LogisticRegression
    • KNeighborsClassifier
    • DecisionTreeClassifier
    • RandomForestClassifier
    • GradientBoostingClassifier
    • SVC
    • GaussianNB
    • BernoulliNB
    • MultinomialNB
  2. Returns a results dataframe that has information of the model name, accuracy and F1-score on the test data.

  3. The package takes the following parameters as input:

    • dataset_path: Path to the csv or excel dataset.
    • output_column: Name of the output column which contains the target variable.
    • train_test_ratio: Ratio in which the dataset is to be divided in train and test splits.

Usage

  • Make sure you have Python installed in your system.
  • Run Following command in the CMD.
    pip install classifier_agent
    

Example

from classifier_agent import classifier_agent

dataset_path = "diabetes.csv"
output_column = "Outcome"
train_test_ratio = 0.25

results = classifier_agent(dataset_path, output_column, train_test_ratio)
print(results)

Note

  • The package is currently in a very elementary stage and work is in progress.
  • The whole project is developed with python version Python 3.7.7 and pip version pip 19.2.3.
  • In case of error, feel free to contact me over Linkedin at Adnan.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

classifier_agent-0.0.2.tar.gz (2.0 kB view hashes)

Uploaded Source

Built Distribution

classifier_agent-0.0.2-py3-none-any.whl (3.0 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page