Skip to content

Latest commit

 

History

History
60 lines (39 loc) · 2.63 KB

README.md

File metadata and controls

60 lines (39 loc) · 2.63 KB

Classification-Report

Documentation Status GitHub License

Classification Report is a high-level library built on top of Pytorch which utilizes Tensorboard and scikit-learn and can be used for any classification problem. It tracks models Weight, Biases and Gradients during training and generates a detailed evaluation report for the model, all of this can be visualized on Tensorboard giving comphrensive insights. It can also be used for HyperParameter tracking which then can be utilized to compare different experiments.

Features

  1. Model Weights, Biases and Gradients Tracking and plotting on histogram.
  2. Visualizing the distribution of above described Model parameters.
  3. Generating an interactive graph of the entire Model.
  4. Graph of Precision, Recall and F1 Score for all the classes for each epoch.
  5. Graph of Macro Avg and Weighted Avg of Precision, Recall and F1-score for each epoch.
  6. Training and Validation Loss tracking for each epoch.
  7. Accuracy and MCC metric tracking at each epoch.
  8. Generating Confusion Matrix after certain number of epochs.
  9. Bar Graph for False Positive and False Negative count for each class.
  10. Scatter Plot for the predicited probabilities.
  11. HyperParameter Tracking for comparing experiments.
pip install classification-report

Goole Colab (Recommended)

​ Just open this notebook on colab and view the entire tensorboard visualization. - Simple Mnist Simple Reporting Visualize on colab

Usage

Understand The Usage

Say Thanks, By connecting me over Linkedin.

SayThanks