No public description

PiperOrigin-RevId: 639874888

lit_nlp/examples/blank_slate_demo.py

@@ -35,13 +35,13 @@
 from lit_nlp.examples.datasets import imagenette
 from lit_nlp.examples.datasets import lm
 from lit_nlp.examples.datasets import mt
-from lit_nlp.examples.datasets import penguin_data
 from lit_nlp.examples.datasets import summarization
 from lit_nlp.examples.models import glue_models
 from lit_nlp.examples.models import mobilenet
-from lit_nlp.examples.models import penguin_model
 from lit_nlp.examples.models import pretrained_lms
 from lit_nlp.examples.models import t5
+from lit_nlp.examples.penguin import data as penguin_data
+from lit_nlp.examples.penguin import model as penguin_model
 
 # NOTE: additional flags defined in server_flags.py
 

lit_nlp/examples/penguin_demo.py → lit_nlp/examples/penguin/demo.py

@@ -1,7 +1,7 @@
 """🐧 LIT demo for tabular data using penguin classification.
 
 To run:
-  python -m lit_nlp.examples.penguin_demo --port=5432
+  blaze run -c opt --config=cuda examples/penguin:demo -- --port=5432
 
 Then navigate to localhost:5432 to access the demo UI.
 """
@@ -17,8 +17,8 @@
 from lit_nlp import server_flags
 from lit_nlp.api import layout
 from lit_nlp.components import minimal_targeted_counterfactuals
-from lit_nlp.examples.datasets import penguin_data
-from lit_nlp.examples.models import penguin_model
+from lit_nlp.examples.penguin import data as penguin_data
+from lit_nlp.examples.penguin import model as penguin_model
 
 MODEL_PATH = 'https://storage.googleapis.com/what-if-tool-resources/lit-models/penguin.h5'  # pylint: disable=line-too-long
 

lit_nlp/examples/models/penguin_model.py → lit_nlp/examples/penguin/model.py

@@ -2,7 +2,7 @@
 
 from lit_nlp.api import model as lit_model
 from lit_nlp.api import types as lit_types
-from lit_nlp.examples.datasets import penguin_data
+from lit_nlp.examples.penguin import data as penguin_data
 from lit_nlp.lib import file_cache
 import numpy as np
 import tensorflow as tf

lit_nlp/examples/models/penguin_model_int_test.py → lit_nlp/examples/penguin/model_int_test.py

@@ -1,7 +1,7 @@
 """Integration tests for penguin_model."""
 
 from absl.testing import absltest
-from lit_nlp.examples.models import penguin_model
+from lit_nlp.examples.penguin import model as penguin_model
 
 
 class PenguinModelIntTest(absltest.TestCase):

website/sphinx_src/components.md

@@ -233,7 +233,7 @@ these cases, LIT validates model compatibility by checking that:
     (`SparseMultilabel`) field.
 
 For a demo using a penguin stats dataset/binary classification task, see
-[examples/penguin_demo.py](https://github.com/PAIR-code/lit/blob/main/lit_nlp/examples/penguin_demo.py).
+[examples/penguin/demo.py](https://github.com/PAIR-code/lit/blob/main/lit_nlp/examples/penguin/demo.py).
 
 ### Images
 

website/sphinx_src/demos.md

@@ -159,7 +159,7 @@ https://pair-code.github.io/lit/tutorials/coref
 ### Tabular Data: Penguin Classification <!-- DO NOT REMOVE {#penguin .demo-header} -->
 
 **Hosted instance:** https://pair-code.github.io/lit/demos/penguins.html \
-**Code:** [examples/penguin_demo.py](https://github.com/PAIR-code/lit/blob/main/lit_nlp/examples/penguin_demo.py)
+**Code:** [examples/penguin/demo.py](https://github.com/PAIR-code/lit/blob/main/lit_nlp/examples/penguin/demo.py)
 
 *   Binary classification on
     [penguin dataset](https://www.tensorflow.org/datasets/catalog/penguins).