wandb 0.16.6__py3-none-any.whl → 0.17.0__py3-none-any.whl

{wandb-0.16.6.dist-info → wandb-0.17.0.dist-info}/WHEEL

@@ -1,5 +1,4 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: hatchling 1.24.2
 Root-Is-Purelib: true
 Tag: py3-none-any
-

wandb/catboost/__init__.py

@@ -1,9 +0,0 @@
-"""Compatibility catboost module.
-
-In the future use:
-    from wandb.integration.catboost import WandbCallback
-"""
-
-from wandb.integration.catboost import WandbCallback, log_summary
-
-__all__ = ["log_summary", "WandbCallback"]

wandb/fastai/__init__.py

@@ -1,9 +0,0 @@
-"""Compatibility fastai module.
-
-In the future use:
-    from wandb.integration.fastai import WandbCallback
-"""
-
-from wandb.integration.fastai import WandbCallback
-
-__all__ = ["WandbCallback"]

wandb/keras/__init__.py

@@ -1,18 +0,0 @@
-"""Compatibility keras module.
-
-In the future use e.g.:
-    from wandb.integration.keras import WandbCallback
-"""
-__all__ = (
-    "WandbCallback",
-    "WandbMetricsLogger",
-    "WandbModelCheckpoint",
-    "WandbEvalCallback",
-)
-
-from wandb.integration.keras import (
-    WandbCallback,
-    WandbEvalCallback,
-    WandbMetricsLogger,
-    WandbModelCheckpoint,
-)

wandb/lightgbm/__init__.py

@@ -1,9 +0,0 @@
-"""Compatibility lightgbm module.
-
-In the future use:
-    from wandb.integration.lightgbm import wandb_callback
-"""
-
-from wandb.integration.lightgbm import log_summary, wandb_callback
-
-__all__ = ["wandb_callback", "log_summary"]

wandb/plots/__init__.py

@@ -1,6 +0,0 @@
-from wandb.plots.explain_text import explain_text as ExplainText
-from wandb.plots.heatmap import heatmap as HeatMap
-from wandb.plots.named_entity import named_entity as NER
-from wandb.plots.part_of_speech import part_of_speech as POS
-from wandb.plots.precision_recall import precision_recall
-from wandb.plots.roc import roc as ROC

wandb/plots/explain_text.py

@@ -1,36 +0,0 @@
-import wandb
-from wandb import util
-from wandb.plots.utils import (
-    deprecation_notice,
-    encode_labels,
-    test_missing,
-    test_types,
-)
-
-
-def explain_text(text, probas, target_names=None):
-    """
-    ExplainText adds support for eli5's LIME based TextExplainer.
-    Arguments:
-     text (str): Text to explain
-     probas (black-box classification pipeline): A function which
-                    takes a list of strings (documents) and returns a matrix
-                    of shape (n_samples, n_classes) with probability values,
-                    i.e. a row per document and a column per output label.
-    Returns:
-     Nothing. To see plots, go to your W&B run page.
-    Example:
-     wandb.log({'roc': wandb.plots.ExplainText(text, probas)})
-    """
-    deprecation_notice()
-    eli5 = util.get_module(
-        "eli5",
-        required="explain_text requires the eli5 library, install with `pip install eli5`",
-    )
-    if test_missing(text=text, probas=probas):
-        # and test_types(proba=proba)):
-        wandb.termlog("Visualizing TextExplainer.")
-        te = eli5.lime.TextExplainer(random_state=42)
-        te.fit(text, probas)
-        html = te.show_prediction(target_names=target_names)
-        return wandb.Html(html.data)

wandb/plots/heatmap.py

@@ -1,81 +0,0 @@
-import wandb
-from wandb import util
-from wandb.plots.utils import (
-    deprecation_notice,
-    encode_labels,
-    test_missing,
-    test_types,
-)
-
-chart_limit = wandb.Table.MAX_ROWS
-
-
-def heatmap(x_labels, y_labels, matrix_values, show_text=False):
-    """
-    Generates a heatmap.
-
-    Arguments:
-     matrix_values (arr): 2D dataset of shape x_labels * y_labels, containing
-                        heatmap values that can be coerced into an ndarray.
-     x_labels  (list): Named labels for rows (x_axis).
-     y_labels  (list): Named labels for columns (y_axis).
-     show_text (bool): Show text values in heatmap cells.
-
-    Returns:
-     Nothing. To see plots, go to your W&B run page then expand the 'media' tab
-           under 'auto visualizations'.
-
-    Example:
-     wandb.log({'heatmap': wandb.plots.HeatMap(x_labels, y_labels,
-                matrix_values)})
-    """
-    deprecation_notice()
-
-    np = util.get_module(
-        "numpy",
-        required="roc requires the numpy library, install with `pip install numpy`",
-    )
-    scikit = util.get_module(
-        "sklearn",
-        required="roc requires the scikit library, install with `pip install scikit-learn`",
-    )
-
-    if test_missing(
-        x_labels=x_labels, y_labels=y_labels, matrix_values=matrix_values
-    ) and test_types(x_labels=x_labels, y_labels=y_labels, matrix_values=matrix_values):
-        matrix_values = np.array(matrix_values)
-        wandb.termlog("Visualizing heatmap.")
-
-        def heatmap_table(x_labels, y_labels, matrix_values, show_text):
-            x_axis = []
-            y_axis = []
-            values = []
-            count = 0
-            for i, x in enumerate(x_labels):
-                for j, y in enumerate(y_labels):
-                    x_axis.append(x)
-                    y_axis.append(y)
-                    values.append(matrix_values[j][i])
-                    count += 1
-                    if count >= chart_limit:
-                        wandb.termwarn(
-                            "wandb uses only the first %d datapoints to create the plots."
-                            % wandb.Table.MAX_ROWS
-                        )
-                        break
-            if show_text:
-                heatmap_key = "wandb/heatmap/v1"
-            else:
-                heatmap_key = "wandb/heatmap_no_text/v1"
-            return wandb.visualize(
-                heatmap_key,
-                wandb.Table(
-                    columns=["x_axis", "y_axis", "values"],
-                    data=[
-                        [x_axis[i], y_axis[i], round(values[i], 2)]
-                        for i in range(len(x_axis))
-                    ],
-                ),
-            )
-
-        return heatmap_table(x_labels, y_labels, matrix_values, show_text)

wandb/plots/named_entity.py

@@ -1,43 +0,0 @@
-import wandb
-from wandb import util
-from wandb.plots.utils import (
-    deprecation_notice,
-    encode_labels,
-    test_missing,
-    test_types,
-)
-
-
-def named_entity(docs):
-    """
-    Adds support for spaCy's entity visualizer, which highlights named
-        entities and their labels in a text.
-
-    Arguments:
-     docs (list, Doc, Span): Document(s) to visualize.
-
-    Returns:
-     Nothing. To see plots, go to your W&B run page.
-
-    Example:
-     wandb.log({'NER': wandb.plots.NER(docs=doc)})
-    """
-    deprecation_notice()
-
-    spacy = util.get_module(
-        "spacy",
-        required="part_of_speech requires the spacy library, install with `pip install spacy`",
-    )
-    en_core_web_md = util.get_module(
-        "en_core_web_md",
-        required="part_of_speech requires the en_core_web_md library, install with `python -m spacy download en_core_web_md`",
-    )
-    nlp = en_core_web_md.load()
-
-    if test_missing(docs=docs):
-        # and test_types(docs=docs)):
-        wandb.termlog("Visualizing named entity recognition.")
-        html = spacy.displacy.render(
-            nlp(str(docs)), style="ent", page=True, minify=True
-        )
-        return wandb.Html(html)

wandb/plots/part_of_speech.py

@@ -1,50 +0,0 @@
-import wandb
-from wandb import util
-from wandb.plots.utils import (
-    deprecation_notice,
-    encode_labels,
-    test_missing,
-    test_types,
-)
-
-
-def part_of_speech(docs):
-    """
-    Adds support for spaCy's dependency visualizer which shows
-        part-of-speech tags and syntactic dependencies.
-
-    Arguments:
-     docs (list, Doc, Span): Document(s) to visualize.
-
-    Returns:
-     Nothing. To see plots, go to your W&B run page.
-
-    Example:
-     wandb.log({'part_of_speech': wandb.plots.POS(docs=doc)})
-    """
-    deprecation_notice()
-
-    spacy = util.get_module(
-        "spacy",
-        required="part_of_speech requires the spacy library, install with `pip install spacy`",
-    )
-    en_core_web_md = util.get_module(
-        "en_core_web_md",
-        required="part_of_speech requires the en_core_web_md library, install with `python -m spacy download en_core_web_md`",
-    )
-    nlp = en_core_web_md.load()
-
-    if test_missing(docs=docs):
-        # and test_types(docs=docs)):
-        wandb.termlog("Visualizing part of speech.")
-        options = {
-            "compact": True,
-            "color": "#1a1c1f",
-            "font": "Source Sans Pro",
-            "collapse_punct": True,
-            "collapse_phrases": True,
-        }
-        html = spacy.displacy.render(
-            nlp(str(docs)), style="dep", minify=True, options=options, page=True
-        )
-        return wandb.Html(html)