search

ShifuML / shifu

An end-to-end machine learning and data mining framework on Hadoop
https://github.com/ShifuML/shifu/wiki
Apache License 2.0
251 stars 109 forks source link

Use `shifu init -t xxx` to specify the algorithm in model config #749

Open Liu-Delin opened 3 years ago

Liu-Delin commented 3 years ago

I try to use WDL and it is uneasy to go through all the steps of shifu init/stats/norm/train, because I need to change lots of configuration in ModelConfig.json.

If we can use command like shifu init -t wdl to generate a model config with WDL configuration, it should be easier to use.

Related code: https://github.com/ShifuML/shifu/blob/master/src/main/java/ml/shifu/shifu/container/obj/ModelConfig.java#L267-L285 https://github.com/ShifuML/shifu/blob/2a3c45976e7c8cee83f4178db299609561ceebb2/src/main/java/ml/shifu/shifu/container/obj/ModelTrainConf.java#L531