[incubator-tvm] branch master created (now 1097ba1)

[tq...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.


      at 1097ba1  filter out error features (#5952)

This branch includes the following new commits:

     new 1097ba1  filter out error features (#5952)

The 1 revisions listed above as "new" are entirely new to this
repository and will be described in separate emails.  The revisions
listed as "add" were already present in the repository and have only
been added to this reference.

[incubator-tvm] 01/01: filter out error features (#5952)

[tq...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git

commit 1097ba1a25a21af84b2e837935fc93b735a652ec
Author: Meteorix <lx...@gmail.com>
AuthorDate: Mon Oct 12 02:36:42 2020 +0800

    filter out error features (#5952)
---
 python/tvm/autotvm/tuner/xgboost_cost_model.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/tvm/autotvm/tuner/xgboost_cost_model.py b/python/tvm/autotvm/tuner/xgboost_cost_model.py
index 7b9df1c..9fd19ec 100644
--- a/python/tvm/autotvm/tuner/xgboost_cost_model.py
+++ b/python/tvm/autotvm/tuner/xgboost_cost_model.py
@@ -245,6 +245,8 @@ class XGBoostCostModel(CostModel):
         else:
             raise RuntimeError("Invalid feature type: " + self.fea_type)
         res = pool.map(feature_extract_func, data)
+        # filter out None output, which is caused by extract exception
+        res = [r for r in res if r is not None]
 
         # filter out feature with different shapes
         fea_len = len(self._get_feature([0])[0])