Merge branch 'refactor-for-docs'

Author: tonyfast

.gitignore

@@ -1,3 +1,4 @@
+benchmarks/figures
 _site
 docs/*.md
 docs/*.html

.pytest_cache/v/cache/lastfailed

@@ -0,0 +1 @@
+{}

.pytest_cache/v/cache/nodeids

@@ -0,0 +1,26 @@
+[
+  "src/importnb/tests/test_importnb.ipynb::test_single_file_with_context",
+  "src/importnb/tests/test_importnb.ipynb::test_from_filename",
+  "src/importnb/tests/test_importnb.ipynb::test_with_doctest",
+  "src/importnb/tests/test_importnb.ipynb::test_from_filename_main",
+  "src/importnb/tests/test_importnb.ipynb::test_parameterize",
+  "src/importnb/tests/test_importnb.ipynb::test_python_file",
+  "src/importnb/tests/test_importnb.ipynb::test_single_file_with_capture",
+  "src/importnb/tests/test_importnb.ipynb::test_capturer",
+  "src/importnb/tests/test_importnb.ipynb::test_single_file_with_lazy",
+  "src/importnb/tests/test_importnb.ipynb::test_single_file_without_context",
+  "src/importnb/tests/test_importnb.ipynb::test_single_file_relative",
+  "src/importnb/tests/test_importnb.ipynb::test_single_with_extension",
+  "src/importnb/tests/test_importnb.ipynb::test_package",
+  "src/importnb/tests/test_importnb.ipynb::test_package_failure",
+  "src/importnb/tests/test_importnb.ipynb::test_package_failure_partial",
+  "src/importnb/tests/test_module.ipynb::testmod",
+  "src/importnb/tests/test_module.ipynb::test_all",
+  "src/importnb/tests/test_unittests.ipynb::TestExtension::test_failure",
+  "src/importnb/tests/test_unittests.ipynb::TestExtension::test_import",
+  "src/importnb/tests/test_unittests.ipynb::TestContext::test_import",
+  "src/importnb/tests/test_unittests.ipynb::TestContext::test_reload_with_context",
+  "src/importnb/tests/test_unittests.ipynb::TestRemote::test_imports",
+  "src/importnb/tests/test_unittests.ipynb::TestPartial::test_exception",
+  "src/importnb/tests/test_unittests.ipynb::TestPartial::test_traceback"
+]

benchmarks/hello.png

@@ -0,0 +1,83 @@
+
+import numpy as np
+import matplotlib.pyplot as plt
+from sklearn.tree import DecisionTreeClassifier
+from ipywidgets import interact
+
+
+def visualize_tree(estimator, X, y, boundaries=True,
+                   xlim=None, ylim=None, ax=None):
+    ax = ax or plt.gca()
+    
+    # Plot the training points
+    ax.scatter(X[:, 0], X[:, 1], c=y, s=30, cmap='viridis',
+               clim=(y.min(), y.max()), zorder=3)
+    ax.axis('tight')
+    ax.axis('off')
+    if xlim is None:
+        xlim = ax.get_xlim()
+    if ylim is None:
+        ylim = ax.get_ylim()
+    
+    # fit the estimator
+    estimator.fit(X, y)
+    xx, yy = np.meshgrid(np.linspace(*xlim, num=200),
+                         np.linspace(*ylim, num=200))
+    Z = estimator.predict(np.c_[xx.ravel(), yy.ravel()])
+
+    # Put the result into a color plot
+    n_classes = len(np.unique(y))
+    Z = Z.reshape(xx.shape)
+    contours = ax.contourf(xx, yy, Z, alpha=0.3,
+                           levels=np.arange(n_classes + 1) - 0.5,
+                           cmap='viridis', clim=(y.min(), y.max()),
+                           zorder=1)
+
+    ax.set(xlim=xlim, ylim=ylim)
+    
+    # Plot the decision boundaries
+    def plot_boundaries(i, xlim, ylim):
+        if i >= 0:
+            tree = estimator.tree_
+        
+            if tree.feature[i] == 0:
+                ax.plot([tree.threshold[i], tree.threshold[i]], ylim, '-k', zorder=2)
+                plot_boundaries(tree.children_left[i],
+                                [xlim[0], tree.threshold[i]], ylim)
+                plot_boundaries(tree.children_right[i],
+                                [tree.threshold[i], xlim[1]], ylim)
+        
+            elif tree.feature[i] == 1:
+                ax.plot(xlim, [tree.threshold[i], tree.threshold[i]], '-k', zorder=2)
+                plot_boundaries(tree.children_left[i], xlim,
+                                [ylim[0], tree.threshold[i]])
+                plot_boundaries(tree.children_right[i], xlim,
+                                [tree.threshold[i], ylim[1]])
+            
+    if boundaries:
+        plot_boundaries(0, xlim, ylim)
+
+
+def plot_tree_interactive(X, y):
+    def interactive_tree(depth=5):
+        clf = DecisionTreeClassifier(max_depth=depth, random_state=0)
+        visualize_tree(clf, X, y)
+
+    return interact(interactive_tree, depth=[1, 5])
+
+
+def randomized_tree_interactive(X, y):
+    N = int(0.75 * X.shape[0])
+    
+    xlim = (X[:, 0].min(), X[:, 0].max())
+    ylim = (X[:, 1].min(), X[:, 1].max())
+    
+    def fit_randomized_tree(random_state=0):
+        clf = DecisionTreeClassifier(max_depth=15)
+        i = np.arange(len(y))
+        rng = np.random.RandomState(random_state)
+        rng.shuffle(i)
+        visualize_tree(clf, X[i[:N]], y[i[:N]], boundaries=False,
+                       xlim=xlim, ylim=ylim)
+    
+    interact(fit_randomized_tree, random_state=[0, 100]);

benchmarks/helpers_05_08.py

@@ -11,7 +11,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "# 0.3.0\n",
+    "# 0.2.9\n",
     "\n",
     "* Include `Partial`, `Lazy`, and `NotebookTest` loaders.\n",
     "* Transform markdown cells to literate block strings so they are included in the ast.\n",