diff --git a/doc/api/api.rst b/doc/api/api.rst index 5844055a..39b35219 100644 --- a/doc/api/api.rst +++ b/doc/api/api.rst @@ -101,6 +101,8 @@ Utility classes utils.DMLDummyClassifier utils.DoubleMLBLP utils.DoubleMLPolicyTree + utils.GlobalRegressor + utils.GlobalClassifier Utility functions ~~~~~~~~~~~~~~~~~ diff --git a/doc/examples/py_double_ml_pension.ipynb b/doc/examples/py_double_ml_pension.ipynb index b713ae1a..251baf7b 100644 --- a/doc/examples/py_double_ml_pension.ipynb +++ b/doc/examples/py_double_ml_pension.ipynb @@ -94,23 +94,6 @@ "data = fetch_401K(return_type='DataFrame')" ] }, - { - "cell_type": "code", - "execution_count": null, - "id": "853ba4be", - "metadata": {}, - "outputs": [], - "source": [ - "# Temporary fix for https://github.com/DoubleML/doubleml-docs/issues/45 / https://github.com/scikit-learn/scikit-learn/issues/21997\n", - "# Can be removed when scikit-learn version 1.2.0 is released\n", - "dtypes = data.dtypes\n", - "dtypes['nifa'] = 'float64'\n", - "dtypes['net_tfa'] = 'float64'\n", - "dtypes['tw'] = 'float64'\n", - "dtypes['inc'] = 'float64'\n", - "data = data.astype(dtypes)" - ] - }, { "cell_type": "code", "execution_count": null, @@ -1179,7 +1162,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.11.2" + "version": "3.12.4" }, "vscode": { "interpreter": {