如何在xgboost中获取pyspark的默认参数?

bkhjykvo  于 2021-07-13  发布在  Spark
关注(0)|答案(0)|浏览(209)

我在用 XGboost4j-0.90Pyspark . 我想获取 XGBoostClassifier() 但我不知道怎么做。这个 Pyspark api似乎有一个方法来做它,但它不工作。例如,如果我尝试:

xgb = XGBoostClassifier()
xgb.getOrDefault('gamma')

我明白了:

KeyErrorTraceback (most recent call last)
<ipython-input-36-7bc03010f512> in <module>()
      1 xgb = XGBoostClassifier()
----> 2 xgb.getOrDefault('gamma')

/opt/cloudera/parcels/CDH-6.3.4-1.cdh6.3.4.p0.6626826/lib/spark/python/pyspark/ml/param/__init__.py in getOrDefault(self, param)
    346             return self._paramMap[param]
    347         else:
--> 348             return self._defaultParamMap[param]
    349 
    350     def extractParamMap(self, extra=None):

KeyError: Param(parent=u'XGBoostClassifier_2bb91ea38c7a', name='gamma', doc='minimum loss reduction required to make a further partition on a leaf node of the tree. the larger, the more conservative the algorithm will be.')

还有另一种方法:

xgb.extractParamMap()

但这只返回您以前传递模型的任何参数,而不是默认参数。如果在创建模型时没有传递任何参数,则此方法将返回一个空dict。
有人知道如何获得这个模型的默认参数吗?

暂无答案!

目前还没有任何答案,快来回答吧!

相关问题