Fix fastai#155

borisdayma · Mar 9, 2020 · 4bde87b · 4bde87b
1 parent 9024c04
commit 4bde87b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 31 deletions.
diff --git a/fastai2/_nbdev.py b/fastai2/_nbdev.py
@@ -460,7 +460,6 @@
          "FBeta": "13b_metrics.ipynb",
          "HammingLoss": "13b_metrics.ipynb",
          "Jaccard": "13b_metrics.ipynb",
-         "MatthewsCorrCoef": "13b_metrics.ipynb",
          "Precision": "13b_metrics.ipynb",
          "Recall": "13b_metrics.ipynb",
          "RocAuc": "13b_metrics.ipynb",
@@ -518,6 +517,8 @@
          "CollectDataCallback": "14a_callback.data.ipynb",
          "WeightedDL": "14a_callback.data.ipynb",
          "weighted_dataloaders": "14a_callback.data.ipynb",
+         "PartialDL": "14a_callback.data.ipynb",
+         "partial_dataloaders": "14a_callback.data.ipynb",
          "Hook": "15_callback.hook.ipynb",
          "hook_output": "15_callback.hook.ipynb",
          "Hooks": "15_callback.hook.ipynb",

diff --git a/fastai2/metrics.py b/fastai2/metrics.py
@@ -1,12 +1,11 @@
 # AUTOGENERATED! DO NOT EDIT! File to edit: nbs/13b_metrics.ipynb (unless otherwise specified).
 
 __all__ = ['AccumMetric', 'skm_to_fastai', 'optim_metric', 'accuracy', 'error_rate', 'top_k_accuracy', 'APScore',
-           'BalancedAccuracy', 'BrierScore', 'CohenKappa', 'F1Score', 'FBeta', 'HammingLoss', 'Jaccard',
-           'MatthewsCorrCoef', 'Precision', 'Recall', 'RocAuc', 'Perplexity', 'perplexity', 'accuracy_multi',
-           'APScoreMulti', 'BrierScoreMulti', 'F1ScoreMulti', 'FBetaMulti', 'HammingLossMulti', 'JaccardMulti',
-           'MatthewsCorrCoefMulti', 'PrecisionMulti', 'RecallMulti', 'RocAucMulti', 'mse', 'rmse', 'mae', 'msle',
-           'exp_rmspe', 'ExplainedVariance', 'R2Score', 'foreground_acc', 'Dice', 'JaccardCoeff', 'LossMetric',
-           'LossMetrics']
+           'BalancedAccuracy', 'BrierScore', 'CohenKappa', 'F1Score', 'FBeta', 'HammingLoss', 'Jaccard', 'Precision',
+           'Recall', 'RocAuc', 'Perplexity', 'perplexity', 'accuracy_multi', 'APScoreMulti', 'BrierScoreMulti',
+           'F1ScoreMulti', 'FBetaMulti', 'HammingLossMulti', 'JaccardMulti', 'MatthewsCorrCoefMulti', 'PrecisionMulti',
+           'RecallMulti', 'RocAucMulti', 'mse', 'rmse', 'mae', 'msle', 'exp_rmspe', 'ExplainedVariance', 'R2Score',
+           'foreground_acc', 'Dice', 'JaccardCoeff', 'LossMetric', 'LossMetrics']
 
 # Cell
 from .data.all import *
@@ -135,11 +134,6 @@ def Jaccard(axis=-1, labels=None, pos_label=1, average='binary', sample_weight=N
     return skm_to_fastai(skm.jaccard_similarity_score, axis=axis,
                          labels=labels, pos_label=pos_label, average=average, sample_weight=sample_weight)
 
-# Cell
-def MatthewsCorrCoef(axis=-1, sample_weight=None):
-    "Matthews correlation coefficient for single-label binary classification problems"
-    return skm_to_fastai(skm.matthews_corrcoef, axis=axis, sample_weight=sample_weight)
-
 # Cell
 def Precision(axis=-1, labels=None, pos_label=1, average='binary', sample_weight=None):
     "Precision for single-label classification problems"

diff --git a/nbs/13b_metrics.ipynb b/nbs/13b_metrics.ipynb
@@ -564,25 +564,6 @@
     "See the [scikit-learn documentation](https://scikit-learn.org/stable/modules/generated/sklearn.metrics.jaccard_score.html#sklearn.metrics.jaccard_score) for more details."
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "#export\n",
-    "def MatthewsCorrCoef(axis=-1, sample_weight=None):\n",
-    "    \"Matthews correlation coefficient for single-label binary classification problems\"\n",
-    "    return skm_to_fastai(skm.matthews_corrcoef, axis=axis, sample_weight=sample_weight)"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "See the [scikit-learn documentation](https://scikit-learn.org/stable/modules/generated/sklearn.metrics.matthews_corrcoef.html#sklearn.metrics.matthews_corrcoef) for more details."
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": null,