Release TF-DF 1.4.0 and YDF 1.5.0

PiperOrigin-RevId: 545230857
tensorflow · Jul 3, 2023 · a3d1e4e · a3d1e4e
1 parent 407772c
commit a3d1e4e
Show file tree

Hide file tree

Showing 7 changed files with 27 additions and 16 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,6 +2,23 @@
 
 ## HEAD
 
+## 1.4.0 - 2023-07-03
+
+### Features
+
+-   Support for multi-task learning.
+-   New tutorial for TF-DF <--> TF.js
+-   Support for uplift modeling in the model inspector.
+-   New tutorial for Uplift modeling.
+-   Bump Bazel version to 6.1.0.
+
+### Fix
+
+-   Regex to generate Bazel workspace.
+-   Remove warning when converting Keras -> YDF.
+-   Fixed default hyperparameter issue Github #172.
+-   Various documentation issues fixed.
+
 ## 1.3.0 - 2023-03-23
 
 ### Features

diff --git a/configure/setup.py b/configure/setup.py
@@ -21,7 +21,7 @@
 from setuptools.command.install import install
 from setuptools.dist import Distribution
 
-_VERSION = "1.3.0"
+_VERSION = "1.4.0"
 
 with open("README.md", "r", encoding="utf-8") as fh:
   long_description = fh.read()

diff --git a/documentation/known_issues.md b/documentation/known_issues.md
@@ -46,7 +46,7 @@ The following table shows the compatibility between
 
 tensorflow_decision_forests | tensorflow
 --------------------------- | ---------------
-1.3.0                       | 2.12.0
+1.3.0 - 1.4.0               | 2.12.0
 1.1.0 - 1.2.0               | 2.11.0
 1.0.0 - 1.0.1               | 2.10.0 - 2.10.1
 0.2.6 - 0.2.7               | 2.9.1

diff --git a/tensorflow_decision_forests/__init__.py b/tensorflow_decision_forests/__init__.py
@@ -51,7 +51,7 @@
 
 """
 
-__version__ = "1.3.0"
+__version__ = "1.4.0"
 __author__ = "Mathieu Guillame-Bert"
 
 compatible_tf_versions = ["2.12.0"]

diff --git a/tensorflow_decision_forests/keras/wrappers_pre_generated.py b/tensorflow_decision_forests/keras/wrappers_pre_generated.py
@@ -1050,10 +1050,12 @@ class GradientBoostedTreesModel(core.CoreModel):
       values are: - `DEFAULT`: Select the loss automatically according to the
       task and label statistics. - `BINOMIAL_LOG_LIKELIHOOD`: Binomial log
       likelihood. Only valid for binary classification. - `SQUARED_ERROR`: Least
-      square loss. Only valid for regression. - `MULTINOMIAL_LOG_LIKELIHOOD`:
-      Multinomial log likelihood i.e. cross-entropy. Only valid for binary or
-      multi-class classification. - `LAMBDA_MART_NDCG5`: LambdaMART with NDCG5.
-      - `XE_NDCG_MART`:  Cross Entropy Loss NDCG. See arxiv.org/abs/1911.09798.
+      square loss. Only valid for regression. - `POISSON`: Poisson log
+      likelihood loss. Mainly used for counting problems. Only valid for
+      regression. - `MULTINOMIAL_LOG_LIKELIHOOD`: Multinomial log likelihood
+      i.e. cross-entropy. Only valid for binary or multi-class classification. -
+      `LAMBDA_MART_NDCG5`: LambdaMART with NDCG5. - `XE_NDCG_MART`:  Cross
+      Entropy Loss NDCG. See arxiv.org/abs/1911.09798.
         Default: "DEFAULT".
     max_depth: Maximum depth of the tree. `max_depth=1` means that all trees
       will be roots. Negative values are ignored. Default: 6.
@@ -1405,8 +1407,6 @@ def capabilities() -> abstract_learner_pb2.LearnerCapabilities:
 class HyperparameterOptimizerModel(core.CoreModel):
   r"""Hyperparameter Optimizer learning algorithm.
 
-
-
   Usage example:
 
   ```python
@@ -1632,12 +1632,9 @@ def capabilities() -> abstract_learner_pb2.LearnerCapabilities:
         support_partial_cache_dataset_format=False
     )
 
-
 class MultitaskerModel(core.CoreModel):
   r"""Multitasker learning algorithm.
 
-
-
   Usage example:
 
   ```python
@@ -1863,7 +1860,6 @@ def capabilities() -> abstract_learner_pb2.LearnerCapabilities:
         support_partial_cache_dataset_format=False
     )
 
-
 class RandomForestModel(core.CoreModel):
   r"""Random Forest learning algorithm.
 

diff --git a/tools/start_compile_docker.sh b/tools/start_compile_docker.sh
@@ -25,7 +25,7 @@
 #  ./tools/start_compile_docker.sh
 #
 #  # Compile and test TF-DF.
-#  RUN_TESTS=1 PY_VERSION=3.8 TF_VERSION=2.11.0 ./tools/test_bazel.sh
+#  RUN_TESTS=1 PY_VERSION=3.8 TF_VERSION=2.12.0 ./tools/test_bazel.sh
 #
 #  # Create a Pip package for a specific version of python.
 #  ./tools/build_pip_package.sh python3.8

diff --git a/tools/test_bazel.sh b/tools/test_bazel.sh
@@ -177,8 +177,6 @@ else
   TEST_RULES="//tensorflow_decision_forests/component/...:all //tensorflow_decision_forests/contrib/...:all //tensorflow_decision_forests/keras/...:all"
 fi
 
-ls -l -a
-
 # Build library
 time ${BAZEL} ${STARTUP_FLAGS} build ${BUILD_RULES} ${FLAGS}