Skip to content

Commit 5b01aba

Browse files
author
minjk-bl
committed
Add model save/load app
1 parent 4bd7e66 commit 5b01aba

7 files changed

Lines changed: 163 additions & 0 deletions

File tree

‎visualpython/css/m_ml/saveLoad.css‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
.vp-popup-frame select.vp-modelio-type {
2+
border: 2px solid #FFCF73;
3+
width: 100%;
4+
height: 35px;
5+
background-position: 97% 50%;
6+
cursor: pointer;
7+
}

‎visualpython/css/menuFrame.css‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -427,6 +427,9 @@ input.vp-menu-search-box {
427427
.vp-menuitem.apps .ml_dimensionReduction {
428428
background: top / contain no-repeat url(../img/apps/apps_dimension.svg);
429429
}
430+
.vp-menuitem.apps .ml_saveLoad {
431+
background: top / contain no-repeat url(../img/apps/apps_file.svg);
432+
}
430433
.vp-menuitem.apps .ml_fitPredict {
431434
background: top / contain no-repeat url(../img/apps/apps_fit.svg);
432435
}

‎visualpython/data/libraries.json‎

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3285,6 +3285,20 @@
32853285
"icon": "apps/apps_dimension.svg"
32863286
}
32873287
},
3288+
{
3289+
"id" : "ml_saveLoad",
3290+
"type" : "function",
3291+
"level": 1,
3292+
"name" : "Save/Load",
3293+
"tag" : "SAVE,LOAD,MACHINE LEARNING,ML",
3294+
"path" : "visualpython - machine_learning - save_load",
3295+
"desc" : "Model save/load for machine learning",
3296+
"file" : "m_ml/SaveLoad",
3297+
"apps" : {
3298+
"color": 8,
3299+
"icon": "apps/apps_file.svg"
3300+
}
3301+
},
32883302
{
32893303
"id" : "ml_fitPredict",
32903304
"type" : "function",

‎visualpython/data/m_ml/mlLibrary.js‎

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -649,6 +649,25 @@ define([
649649
{ name: 'learning_rate', component: ['input_number'], default: 200.0, usePair: true },
650650
{ name: 'random_state', component: ['input_number'], placeholder: '123', usePair: true }
651651
]
652+
},
653+
/** Save/Load */
654+
'model_save': {
655+
name: 'Save model',
656+
import: 'import joblib',
657+
code: "joblib.dump(${target}, ${savePath})",
658+
options: [
659+
{ name: 'target', component: ['data_select'], placeholder: 'Select model'},
660+
{ name: 'savePath', component: ['file-save'], placeholder: 'Select path'}
661+
]
662+
},
663+
'model_load': {
664+
name: 'Load model',
665+
import: 'import joblib',
666+
code: "${allocateTo} = joblib.load(${loadPath})",
667+
options: [
668+
{ name: 'loadPath', component: ['file-open'], placeholder: 'Select path'},
669+
{ name: 'allocateTo', component: ['data_select'], placeholder: 'New variable to load'}
670+
]
652671
}
653672
}
654673

‎visualpython/html/m_ml/saveLoad.html‎

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
<body>
2+
<div>
3+
<select id="modelio" class="vp-modelio-type">
4+
<option value="model_save">Save model</option>
5+
<option value="model_load">Load model</option>
6+
</select>
7+
<div class="vp-modelio-option-box vp-grid-border-box vp-grid-col-110 mt5">
8+
9+
</div>
10+
</div>
11+
</body>

‎visualpython/js/com/com_Config.js‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -187,6 +187,10 @@ define([
187187
'Counter': {
188188
code: 'from collections import Counter',
189189
type: 'package'
190+
},
191+
'joblib': {
192+
code: 'import joblib',
193+
type: 'package'
190194
}
191195
}
192196

‎visualpython/js/m_ml/SaveLoad.js‎

Lines changed: 105 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,105 @@
1+
/*
2+
* Project Name : Visual Python
3+
* Description : GUI-based Python code generator
4+
* File Name : SaveLoad.js
5+
* Author : Black Logic
6+
* Note : Save and load models
7+
* License : GNU GPLv3 with Visual Python special exception
8+
* Date : 2023. 03. 09
9+
* Change Date :
10+
*/
11+
12+
//============================================================================
13+
// [CLASS] DataSets
14+
//============================================================================
15+
define([
16+
__VP_CSS_LOADER__('vp_base/css/m_ml/saveLoad'),
17+
__VP_TEXT_LOADER__('vp_base/html/m_ml/saveLoad.html'),
18+
'vp_base/js/com/com_util',
19+
'vp_base/js/com/com_Const',
20+
'vp_base/js/com/com_String',
21+
'vp_base/js/com/com_generatorV2',
22+
'vp_base/js/com/component/PopupComponent',
23+
'vp_base/js/com/component/FileNavigation',
24+
'vp_base/data/m_ml/mlLibrary',
25+
], function(slCss, slHTML, com_util, com_Const, com_String, com_generator, PopupComponent, FileNavigation, ML_LIBRARIES) {
26+
27+
/**
28+
* SaveLoad
29+
*/
30+
class SaveLoad extends PopupComponent {
31+
_init() {
32+
super._init();
33+
this.config.sizeLevel = 2;
34+
this.config.dataview = false;
35+
this.config.checkModules = ['joblib'];
36+
37+
this.state = {
38+
modelio: 'model_save', // model_save / model_load
39+
target: '',
40+
allocateTo: '',
41+
path: '',
42+
...this.state
43+
}
44+
45+
this.mlConfig = ML_LIBRARIES;
46+
}
47+
48+
_bindEvent() {
49+
super._bindEvent();
50+
let that = this;
51+
52+
// select model
53+
$(this.wrapSelector('#modelio')).on('change', function() {
54+
let modelio = $(this).val();
55+
that.state.modelio = modelio;
56+
$(that.wrapSelector('.vp-modelio-option-box')).html(that.templateForOption(modelio));
57+
});
58+
}
59+
60+
templateForBody() {
61+
let page = $(slHTML);
62+
63+
// render option page
64+
$(page).find('.vp-modelio-option-box').html(this.templateForOption(this.state.modelio));
65+
66+
return page;
67+
}
68+
69+
templateForOption(modelio) {
70+
let config = this.mlConfig[modelio];
71+
let state = this.state;
72+
73+
let optBox = new com_String();
74+
// render tag
75+
config.options.forEach(opt => {
76+
optBox.appendFormatLine('<label for="{0}" title="{1}">{2}</label>'
77+
, opt.name, opt.name, opt.name);
78+
let content = com_generator.renderContent(this, opt.component[0], opt, state);
79+
optBox.appendLine(content[0].outerHTML);
80+
});
81+
82+
// render file navigation
83+
84+
85+
// show user option
86+
if (config.code.includes('${etc}')) {
87+
// render user option
88+
optBox.appendFormatLine('<label for="{0}">{1}</label>', 'userOption', 'User option');
89+
optBox.appendFormatLine('<input type="text" class="vp-input vp-state" id="{0}" placeholder="{1}" value="{2}"/>',
90+
'userOption', 'key=value, ...', this.state.userOption);
91+
}
92+
return optBox.toString();
93+
}
94+
95+
generateCode() {
96+
let { modelio, userOption } = this.state;
97+
let code = new com_String();
98+
let modelCode = com_generator.vp_codeGenerator(this, this.mlConfig[modelio], this.state);
99+
code.append(modelCode);
100+
return code.toString();
101+
}
102+
}
103+
104+
return SaveLoad;
105+
});

0 commit comments

Comments
 (0)