Skip to content

SPARK-1544 Add support for deep decision trees. #475

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
manishamde wants to merge 22 commits into from
Closed

SPARK-1544 Add support for deep decision trees. #475

Changes from 1 commit
Commits
Show all changes
22 commits
Select commit Hold shift + click to select a range
50b143a
adding support for very deep trees
manishamde Apr 20, 2014
abc5a23
Parameterizing max memory.
etrain Apr 22, 2014
2f6072c
Merge pull request #5 from etrain/deep_tree
manishamde Apr 22, 2014
2f1e093
minor: added doc for maxMemory parameter
manishamde Apr 22, 2014
0287772
Fixing scalastyle issue.
etrain Apr 22, 2014
fecf89a
Merge pull request #6 from etrain/deep_tree
manishamde Apr 22, 2014
719d009
updating user documentation
manishamde Apr 24, 2014
9dbdabe
merge from master
manishamde Apr 29, 2014
1517155
updated documentation
manishamde Apr 29, 2014
718506b
added unit test
manishamde Apr 30, 2014
e0426ee
renamed parameter
manishamde May 1, 2014
dad9652
removed unused imports
manishamde May 1, 2014
cbd9f14
modified scala.math to math
manishamde May 3, 2014
5e82202
added documentation, fixed off by 1 error in max level calculation
manishamde May 6, 2014
4731cda
formatting
manishamde May 6, 2014
5eca9e4
grammar
manishamde May 6, 2014
8053fed
more formatting
manishamde May 6, 2014
426bb28
programming guide blurb
manishamde May 6, 2014
b27ad2c
formatting
manishamde May 6, 2014
ce004a1
minor formatting
manishamde May 6, 2014
7fc9545
added docs
manishamde May 7, 2014
968ca9d
merged master
manishamde May 7, 2014
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
more formatting
  • Loading branch information
@manishamde
manishamde committed May 6, 2014
commit 8053fed22249bc788ba988489caa22f732b6416d
6 changes: 3 additions & 3 deletions mllib/src/test/scala/org/apache/spark/mllib/tree/DecisionTreeSuite.scala
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -405,9 +405,9 @@ class DecisionTreeSuite extends FunSuite with BeforeAndAfterAll {
assert(splits(0).length === 99)
assert(bins(0).length === 100)

val leftFilter = Filter(new Split(0, 400, FeatureType.Continuous,List()), -1)
val rightFilter = Filter(new Split(0, 400, FeatureType.Continuous,List()) ,1)
val filters = Array[List[Filter]](List(),List(leftFilter),List(rightFilter))
val leftFilter = Filter(new Split(0, 400, FeatureType.Continuous, List()), -1)
val rightFilter = Filter(new Split(0, 400, FeatureType.Continuous, List()) ,1)
val filters = Array[List[Filter]](List(),List(leftFilter), List(rightFilter))
Copy link
Contributor

@mengxr mengxr May 6, 2014

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Forgot one space.

val parentImpurities = Array(0.5, 0.5, 0.5)

// Single group second level tree construction.
Expand Down