Skip to main content

Advertisement

Table 3 GBT tree

From: Analysis and classification of heart diseases using heartbeat features and machine learning algorithms

var   dTree-am   \(=\) new   GBTClassifier()
.setLabelCol("label")//beat type
.setFeaturesCol("features")//chosen features
.setMaxDepth(20)//depth of GBT
.setMaxIter(15)//iteration of GBT