easy_ml 0.2.0.pre.rc48 → 0.2.0.pre.rc49

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 528284a7ca2b0c4aa7370104e5d3e6206b6ada6785da1f76274bb8ce01e74145
-  data.tar.gz: c9aa88ad10369733a124142ed47d1595a06986be06c49817c1b08142992d7146
+  metadata.gz: e342ccd144abd65ab937bace40c11524e76a82aef6a6c5eb16d3ba0b3eae1e84
+  data.tar.gz: f90fc476542773a25d4f8e1995f68b5ceb733d5a9b03e39f687d9a157aade66d
 SHA512:
-  metadata.gz: 8c37ec1e30a93c91594f305078d7282d0aa5e51cddce120d983e32ff34a04633aad0fd388538314d724d2ae7cb2adb5da58bd65b881aa10a5b94c07f2faca32a
-  data.tar.gz: abe7ad4855b7cae96535cd828e47886ed25553b2a7d396811e6846857ccf31c402e5f9536ddd60577ab91bfcc7b538810a05db6622bd1c81e2ba20390c506fb6
+  metadata.gz: 638190fdf3351bb87f44c6276c9159abeb4a1a12cf135942c888a365aba76b21a0a6b5f43e5adfd44908fec1c2ef3cfe5fa7010c4ba964052de8c6948601d959
+  data.tar.gz: 36a4b4d3a34fc51f7283a9d1d2d517e0c09f040286070abda480181ebdf012c5ae158f4f6e9f79d6da53b22692fe675094e899bb32a21da1f4b1f6f307332455

data/app/frontend/components/ScheduleModal.tsx

@@ -44,16 +44,16 @@ interface ScheduleModalProps {
 
 const METRICS = {
   classification: [
-    { value: 'accuracy_score', label: 'Accuracy', description: 'Overall prediction accuracy' },
-    { value: 'precision_score', label: 'Precision', description: 'Ratio of true positives to predicted positives' },
-    { value: 'recall_score', label: 'Recall', description: 'Ratio of true positives to actual positives' },
-    { value: 'f1_score', label: 'F1 Score', description: 'Harmonic mean of precision and recall' }
+    { value: 'accuracy_score', label: 'Accuracy', description: 'Overall prediction accuracy', direction: 'maximize' },
+    { value: 'precision_score', label: 'Precision', description: 'Ratio of true positives to predicted positives', direction: 'maximize' },
+    { value: 'recall_score', label: 'Recall', description: 'Ratio of true positives to actual positives', direction: 'maximize' },
+    { value: 'f1_score', label: 'F1 Score', description: 'Harmonic mean of precision and recall', direction: 'maximize' }
   ],
   regression: [
-    { value: 'mean_absolute_error', label: 'Mean Absolute Error', description: 'Average absolute differences between predicted and actual values' },
-    { value: 'mean_squared_error', label: 'Mean Squared Error', description: 'Average squared differences between predicted and actual values' },
-    { value: 'root_mean_squared_error', label: 'Root Mean Squared Error', description: 'Square root of mean squared error' },
-    { value: 'r2_score', label: 'R² Score', description: 'Proportion of variance in the target that is predictable' }
+    { value: 'mean_absolute_error', label: 'Mean Absolute Error', description: 'Average absolute differences between predicted and actual values', direction: 'minimize' },
+    { value: 'mean_squared_error', label: 'Mean Squared Error', description: 'Average squared differences between predicted and actual values', direction: 'minimize' },
+    { value: 'root_mean_squared_error', label: 'Root Mean Squared Error', description: 'Square root of mean squared error', direction: 'minimize' },
+    { value: 'r2_score', label: 'R² Score', description: 'Proportion of variance in the target that is predictable', direction: 'maximize' }
   ]
 };
 
@@ -609,8 +609,13 @@ export function ScheduleModal({ isOpen, onClose, onSave, initialData, tunerJobCo
                           <div className="ml-3">
                             <h3 className="text-sm font-medium text-blue-800">Deployment Criteria</h3>
                             <p className="mt-2 text-sm text-blue-700">
-                              The model will be automatically deployed when the {formData.retraining_job_attributes.metric} is{' '}
-                              {formData.retraining_job_attributes.direction === 'minimize' ? 'below' : 'above'} {formData.retraining_job_attributes.threshold}.
+                              {(() => {
+                                const metricsList = METRICS[initialData.task === 'classification' ? 'classification' : 'regression'];
+                                const selectedMetric = metricsList.find(m => m.value === formData.retraining_job_attributes.metric);
+                                const direction = selectedMetric?.direction === 'minimize' ? 'below' : 'above';
+                                
+                                return `The model will be automatically deployed when the ${selectedMetric?.label} is ${direction} ${formData.retraining_job_attributes.threshold}.`;
+                              })()}
                             </p>
                           </div>
                         </div>

data/app/jobs/easy_ml/training_job.rb

@@ -10,13 +10,13 @@ module EasyML
 
       @last_activity = Time.current
       setup_signal_traps
-      # @monitor_thread = start_monitor_thread
+      @monitor_thread = start_monitor_thread
 
       @model.actually_train do |iteration_info|
         @last_activity = Time.current
       end
     ensure
-      #   @monitor_thread&.exit
+      @monitor_thread&.exit
     end
 
     private
@@ -42,6 +42,7 @@ module EasyML
       @cleaned_up = true
       @model.last_run.update(status: "failed", error_message: error_message, completed_at: Time.current)
       @model.update(is_training: false)
+      @model.unlock!
     end
 
     def start_monitor_thread

data/app/models/easy_ml/datasource.rb

@@ -46,7 +46,6 @@ module EasyML
     validates :name, presence: true
     validates :datasource_type, presence: true
     validates :datasource_type, inclusion: { in: DATASOURCE_NAMES }
-    # validate :validate_datasource_exists
 
     before_save :set_root_dir
     after_initialize :read_adapter_from_configuration, if: :persisted?

data/lib/easy_ml/version.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module EasyML
-  VERSION = "0.2.0-rc48"
+  VERSION = "0.2.0-rc49"
 
   module Version
   end

data/public/easy_ml/assets/.vite/manifest.json

@@ -1,6 +1,6 @@
 {
   "entrypoints/Application.tsx": {
-    "file": "assets/entrypoints/Application.tsx-KJwHDm3F.js",
+    "file": "assets/entrypoints/Application.tsx-u6L4jkNy.js",
     "name": "entrypoints/Application.tsx",
     "src": "entrypoints/Application.tsx",
     "isEntry": true,