From b69b3c36e218b5071b2d9cca23148e21febffce2 Mon Sep 17 00:00:00 2001
From: Hiroyuki Eguchi <h-eguchi@az.jp.nec.com>
Date: Mon, 13 Feb 2017 14:39:46 +0900
Subject: [PATCH] Add NaiveBayes Model

Change-Id: Ia8f0a96943a66b5e16e19ad666e4c0ee169ad71d
---
 .../machine_learning/models/create/model-model.js  |  1 +
 .../machine_learning/models/create/spec/spec.html  | 14 +++++++++++++-
 .../models/create/spec/template.spec.controller.js |  1 +
 3 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/meteos_ui/static/dashboard/machine_learning/models/create/model-model.js b/meteos_ui/static/dashboard/machine_learning/models/create/model-model.js
index 84e0528..fb98e04 100644
--- a/meteos_ui/static/dashboard/machine_learning/models/create/model-model.js
+++ b/meteos_ui/static/dashboard/machine_learning/models/create/model-model.js
@@ -62,6 +62,7 @@
 
       model.newParamsSpec = {
         numIterations: null,
+        lambda: null,
         numClasses: null,
         runs: null,
         mode: null,
diff --git a/meteos_ui/static/dashboard/machine_learning/models/create/spec/spec.html b/meteos_ui/static/dashboard/machine_learning/models/create/spec/spec.html
index d38000b..cd974be 100644
--- a/meteos_ui/static/dashboard/machine_learning/models/create/spec/spec.html
+++ b/meteos_ui/static/dashboard/machine_learning/models/create/spec/spec.html
@@ -13,7 +13,7 @@
       </div>
     </div>
     <div class="col-xs-12"
-         ng-if="model.newModelSpec.model_type != 'DecisionTreeRegression'">
+         ng-if="model.newModelSpec.model_type != 'DecisionTreeRegression' && model.newModelSpec.model_type != 'NaiveBayes'">
       <div class="form-group">
         <label class="control-label" for="model-model-numIterations">
           <translate>NumIterations</translate>
@@ -24,6 +24,18 @@
                 placeholder="{$ 'Parameters when creating model.'|translate $}">
       </div>
     </div>
+    <div class="col-xs-12"
+         ng-if="model.newModelSpec.model_type == 'NaiveBayes'">
+      <div class="form-group">
+        <label class="control-label" for="model-model-lambda">
+          <translate>Lambda</translate>
+          <span class="hz-icon-required fa fa-asterisk"></span>
+        </label>
+        <input name="model-model-lambda" type="text" class="form-control" id="model-model-lambda"
+               ng-model="model.newParamsSpec.lambda"
+                placeholder="{$ 'Parameters when creating model.'|translate $}">
+      </div>
+    </div>
     <div class="col-xs-12"
          ng-if="model.newModelSpec.model_type == 'KMeans'">
       <div class="form-group">
diff --git a/meteos_ui/static/dashboard/machine_learning/models/create/spec/template.spec.controller.js b/meteos_ui/static/dashboard/machine_learning/models/create/spec/template.spec.controller.js
index cdced37..43583ca 100644
--- a/meteos_ui/static/dashboard/machine_learning/models/create/spec/template.spec.controller.js
+++ b/meteos_ui/static/dashboard/machine_learning/models/create/spec/template.spec.controller.js
@@ -36,6 +36,7 @@
 
     ctrl.modelTypeOptions = [
       'LogisticRegression',
+      'NaiveBayes',
       'DecisionTreeRegression',
       'LinearRegression',
       'KMeans',