Skip to content

Expose options for supervised learning models selection

Latest
Compare
Choose a tag to compare
@dvitsios dvitsios released this 29 Apr 15:13
· 18 commits to master since this release
f0300d3

Major

  • Add option for 'fast' run (-f / --fast) with 4 ML models (instead of the default 6 models)

  • Add option (-m) to explicitly specify the supervised models to be trained by mantis-ml.
    Available model options are:

    • et: Extra Trees
    • rf: Random Forest
    • gb: Gradient Boosting
    • xgb: XGBoost
    • svc: Support Vector Classifier
    • dnn: Deep Neural Net
    • stack: Stacking classifier

    Multiple models may be specified using a ',' separator, e.g. -m et,rf,stack

  • Stacking classifier may now be run using the -m option with the stack arg, i.e.: -m stack

Minor

  • Fix bug in hypergeom_enrichment module when reading external ranked files (only tab-delimiter is currently allowed in case of an external file with two columns)