Skip to content

Commit

Permalink
docstring update and message of not two classes are used
Browse files Browse the repository at this point in the history
  • Loading branch information
erdogant committed May 2, 2020
1 parent c023888 commit 1319dc0
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 5 deletions.
8 changes: 4 additions & 4 deletions classeval/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,16 +15,16 @@

__author__ = 'Erdogan Tasksen'
__email__ = '[email protected]'
__version__ = '0.1.4'
__version__ = '0.1.5'

# module level doc-string
__doc__ = """
classeval
=====================================================================
=============================
Description
-----------
classeval is a python package that contains functionalities and plots for fast and easy classifier evaluation.
classeval is a python package for fast and easy classifier evaluation.
Example
Expand All @@ -48,7 +48,7 @@
References
----------
* https://erdogant.github.io/classeval/
* https://erdogant.github.io/classeval
* https://github.com/erdogant/classeval
* https://scikit-learn.org/stable/modules/model_evaluation.html#scoring-parameter
* http://arogozhnikov.github.io/2015/10/05/roc-curve.html
Expand Down
3 changes: 2 additions & 1 deletion classeval/classeval.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
"""classeval contains functionalities and plots for fast and easy classifier evaluation."""
"""classeval is for fast and easy classifier evaluation."""
# -----------------------------------------------------
# Name : classeval.py
# Author : E.Taskesen
Expand Down Expand Up @@ -122,6 +122,7 @@ def eval(y_true, y_proba, y_score=None, y_pred=None, pos_label=None, threshold=0
elif len(class_names)>2:
out = eval_multiclass(y_true, y_proba, y_score, y_pred, normalize=normalize, verbose=verbose)
else:
print('[classeval] The input variable [y_true] should contain at least two classes.')
out=None

# Return
Expand Down

0 comments on commit 1319dc0

Please sign in to comment.