Extension to uncensored outputs - Githubissues

aldro61 / mmit

Regression trees for interval censored output data

https://aldro61.github.io/mmit/

GNU General Public License v3.0

7 stars 7 forks source link

Extension to uncensored outputs #13

Closed aldro61 closed 7 years ago

aldro61 commented 7 years ago

Code modifications

Uncensored outputs are detected by checking if the target interval bounds are equal
The margin is not used for uncensored outputs
Computing function coefficients and breakpoint positions has been moved out of the PiecewiseFunction class and is now done by the solver.

Unit tests

If all the outputs are uncensored and we use the linear hinge loss, the solution is the median
If all the outputs are uncensored and we use the squared hinge loss, the solution is the mean
Modified the random testing function to include uncensored outputs