机器之心|一行代码不用写,就可以训练、测试、使用模型,这个项目帮你做到( 五 )


# model definitionmodel:type: classificationalgorithm: RandomForestarguments:# notice that this is the available args for the random forest model. check different available args for all supported models by running igel helpn_estimators: 100max_depth: 20
# target you want to predicttarget:- sick
然后 , 可以通过运行 igel 命令来拟合模型:
$ igel fit -dp path_to_the_dataset -yml path_to_the_yaml_file
评估:
$ igel evaluate -dp path_to_the_evaluation_dataset
预测:
$ igel predict -dp path_to_the_new_dataset
参考链接:https://medium.com/@nidhalbacc/machine-learning-without-writing-code-984b238dd890


推荐阅读