easy_ml 0.2.0.pre.rc48 → 0.2.0.pre.rc50

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 528284a7ca2b0c4aa7370104e5d3e6206b6ada6785da1f76274bb8ce01e74145
-  data.tar.gz: c9aa88ad10369733a124142ed47d1595a06986be06c49817c1b08142992d7146
+  metadata.gz: 9571100b7780c1c76d4305566045b890cde5b7a4a0b4ad8526f26d8a3c6337aa
+  data.tar.gz: 129ba88803cfa45fc6382599e0e8615977567c461d2de9b70318f43d45a496ba
 SHA512:
-  metadata.gz: 8c37ec1e30a93c91594f305078d7282d0aa5e51cddce120d983e32ff34a04633aad0fd388538314d724d2ae7cb2adb5da58bd65b881aa10a5b94c07f2faca32a
-  data.tar.gz: abe7ad4855b7cae96535cd828e47886ed25553b2a7d396811e6846857ccf31c402e5f9536ddd60577ab91bfcc7b538810a05db6622bd1c81e2ba20390c506fb6
+  metadata.gz: bb6d4c3b678e58f813a8cc1822a75812fdeb0d2ad65b8d67ecb600bdf2822e522a3cae04c067ad91d4894e6f60dcf3e72bbe52a4206ac7892dde3ad0da8f9a89
+  data.tar.gz: 856438e590cec6306d64c64f7314b81c3eee907fc760fa1fd2978a0fff13fa71df5bc483aae3ee24a5ea7403043f6258f16c985e7e1e334c1e2782822786c56d

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,14 @@ 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
+      @model.unlock!
     end
 
     private
@@ -42,6 +43,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/app/models/easy_ml/retraining_job.rb

@@ -67,6 +67,7 @@ module EasyML
     validate :evaluator_must_be_valid
     validate :validate_at_format
     after_initialize :set_direction, unless: :persisted?
+    before_save :set_direction
 
     scope :active, -> { joins(:model).where(active: true) }
 

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-rc50"
 
   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,