search

HealthCatalyst / healthcareai-py

Python tools for healthcare machine learning
http://healthcare.ai
MIT License
309 stars 186 forks source link

#367, #447 (Change to Unit Tests For #447) #447

Closed danwellisch1 closed 6 years ago

danwellisch1 commented 6 years ago

Hello:

This is a fix for the grain column issue. I just noticed that the description of the issue refers to a missing target column. I didn't do anything with that part of it. Nor did I give the user a message for the missing grain column. If you want either of these, then I can submit another code change. In the meantime, this code change will allow the user to leave out a grain column in training their model, and then on the prediction code while using the model, it does not blow up (expecting the grain column). I tested against the example classification 1 and 2 programs using a grain column and not using a grain column.

Let me know if you want the additional pieces for this issue.....

Dan

danwellisch1 commented 6 years ago

Removed forked repository