1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
import pickle
from Orange.widgets.widget import Input
from Orange.base import Model
from Orange.widgets.utils.save.owsavebase import OWSaveBase
from Orange.widgets.utils.widgetpreview import WidgetPreview
class OWSaveModel(OWSaveBase):
name = "Save Model"
description = "Save a trained model to an output file."
icon = "icons/SaveModel.svg"
replaces = ["Orange.widgets.classify.owsaveclassifier.OWSaveClassifier"]
priority = 3000
keywords = "save model, save"
class Inputs:
model = Input("Model", Model)
filters = ["Pickled model (*.pkcls)"]
@Inputs.model
def set_model(self, model):
self.data = model
self.on_new_input()
def do_save(self):
with open(self.filename, "wb") as f:
pickle.dump(self.data, f)
if __name__ == "__main__": # pragma: no cover
WidgetPreview(OWSaveModel).run()
|
