Return-Path: X-Original-To: archive-asf-public-internal@cust-asf2.ponee.io Delivered-To: archive-asf-public-internal@cust-asf2.ponee.io Received: from cust-asf.ponee.io (cust-asf.ponee.io [163.172.22.183]) by cust-asf2.ponee.io (Postfix) with ESMTP id E3A9B200C1C for ; Wed, 11 Jan 2017 03:03:14 +0100 (CET) Received: by cust-asf.ponee.io (Postfix) id E271C160B51; Wed, 11 Jan 2017 02:03:14 +0000 (UTC) Delivered-To: archive-asf-public@cust-asf.ponee.io Received: from mail.apache.org (hermes.apache.org [140.211.11.3]) by cust-asf.ponee.io (Postfix) with SMTP id 38646160B3D for ; Wed, 11 Jan 2017 03:03:14 +0100 (CET) Received: (qmail 98503 invoked by uid 500); 11 Jan 2017 02:03:13 -0000 Mailing-List: contact reviews-help@spark.apache.org; run by ezmlm Precedence: bulk List-Help: List-Unsubscribe: List-Post: List-Id: Delivered-To: mailing list reviews@spark.apache.org Received: (qmail 98460 invoked by uid 99); 11 Jan 2017 02:03:13 -0000 Received: from git1-us-west.apache.org (HELO git1-us-west.apache.org) (140.211.11.23) by apache.org (qpsmtpd/0.29) with ESMTP; Wed, 11 Jan 2017 02:03:13 +0000 Received: by git1-us-west.apache.org (ASF Mail Server at git1-us-west.apache.org, from userid 33) id 18531DFB7D; Wed, 11 Jan 2017 02:03:13 +0000 (UTC) From: sethah To: reviews@spark.apache.org Reply-To: reviews@spark.apache.org References: In-Reply-To: Subject: [GitHub] spark pull request #16441: [SPARK-14975][ML] Fixed GBTClassifier to predict ... Content-Type: text/plain Message-Id: <20170111020313.18531DFB7D@git1-us-west.apache.org> Date: Wed, 11 Jan 2017 02:03:13 +0000 (UTC) archived-at: Wed, 11 Jan 2017 02:03:15 -0000 Github user sethah commented on a diff in the pull request: https://github.com/apache/spark/pull/16441#discussion_r95495249 --- Diff: mllib/src/main/scala/org/apache/spark/ml/classification/GBTClassifier.scala --- @@ -275,18 +316,33 @@ class GBTClassificationModel private[ml]( @Since("2.0.0") lazy val featureImportances: Vector = TreeEnsembleModel.featureImportances(trees, numFeatures) + private def margin(features: Vector): Double = { --- End diff -- minor: add doc `/** Raw prediction for each class label */` --- If your project is set up for it, you can reply to this email and have your reply appear on GitHub as well. If your project does not have this feature enabled and wishes so, or if the feature is enabled but not working, please contact infrastructure at infrastructure@apache.org or file a JIRA ticket with INFRA. --- --------------------------------------------------------------------- To unsubscribe, e-mail: reviews-unsubscribe@spark.apache.org For additional commands, e-mail: reviews-help@spark.apache.org