Save scikit learn pipeline to external location Parameters ---------- pipelines : sklearn pipeline pipeline to be saved location : str external folder to save pipeline
(self,pipeline,location)
| 165 | # |
| 166 | |
| 167 | def save_pipeline(self,pipeline,location): |
| 168 | """ |
| 169 | Save scikit learn pipeline to external location |
| 170 | |
| 171 | Parameters |
| 172 | ---------- |
| 173 | |
| 174 | pipelines : sklearn pipeline |
| 175 | pipeline to be saved |
| 176 | location : str |
| 177 | external folder to save pipeline |
| 178 | """ |
| 179 | self.create_work_folder() |
| 180 | pipeline_folder = self.work_folder+"/pipeline" |
| 181 | if not os.path.exists(pipeline_folder): |
| 182 | logger.info("creating folder %s",pipeline_folder) |
| 183 | os.makedirs(pipeline_folder) |
| 184 | tmp_file = pipeline_folder+"/p" |
| 185 | joblib.dump(pipeline,tmp_file) |
| 186 | futil = fu.FileUtil(aws_key=self.aws_key,aws_secret=self.aws_secret) |
| 187 | futil.copy(pipeline_folder,location) |
| 188 | |
| 189 | |
| 190 | def load_pipeline(self,pipeline_folder): |