easy_ml 0.2.0.pre.rc38 → 0.2.0.pre.rc39

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cc708b35e777f41f67dbf4e769db3c326b01e1e65f29f293c9536d625a60a197
-  data.tar.gz: c2ba6d165128c9cbeae2cef19ddc40e823208cc75750aa31d1e167fe8c623cd8
+  metadata.gz: a53f346d534b4333dcc9f8880c9d0fd4d14acf7a596be57caf42ea789490e4e4
+  data.tar.gz: 912b118a2c82f2397afce24d39a5c556d98fc58a310647de7232851f7ee606b4
 SHA512:
-  metadata.gz: '07835e0557a01b1db47cbbddcd9149813b0cbbd7083eb951613fa31b32234b5c60a348bca5e3a57446ad21fad017882bf9262a4a2f81c4a6962f6bbd1771de0f'
-  data.tar.gz: 427e065c5b4e83161c030738aa891ad855c6ca89e8b14d729e6602a0285682ac04712b62e5e226dc521bce05f292f783e0728a9c4a517e3aa8417a632b7b5e65
+  metadata.gz: 4d641839d982e782d5921cc7086ef583f3ae0f5901446d0c132deacb5a7101b420428cbbb52f58b0a4f9a94fb3ed6cd7f72cb1705eb2c7c3c4f938c4a7e9702d
+  data.tar.gz: b19d313ece15cb343138f4d6e036947ba764708b80695ef6d7b502c45e3e34662dec959d8e1c1f423109f89ac87101b3383d64b80da3d32cc7e9bee7b30c6f5c

data/app/controllers/easy_ml/predictions_controller.rb

@@ -3,12 +3,23 @@ module EasyML
     skip_before_action :verify_authenticity_token, only: [:create]
 
     def create
+      unless params.key?(:input)
+        return render json: { error: "Must provide key: input" }, status: :not_found
+      end
+      input = params[:input].permit!.to_h
+
+      unless input.is_a?(Hash)
+        return render json: { error: "Input must be a hash" }, status: :not_found
+      end
+
       model_name = params[:model]
-      input = params[:input]
+      unless EasyML::Model.find_by(name: model_name).present?
+        return render json: { error: "Model not found" }, status: :not_found
+      end
 
-      predictions = EasyML::Predict.predict(model_name, input)
+      prediction = EasyML::Predict.predict(model_name, input)
 
-      render json: { prediction: predictions.first }
+      render json: { prediction: EasyML::PredictionSerializer.new(prediction).serializable_hash.dig(:data, :attributes) }, status: :ok
     rescue ActiveRecord::RecordNotFound
       render json: { error: "Model not found" }, status: :not_found
     rescue StandardError => e

data/app/serializers/easy_ml/prediction_serializer.rb

@@ -0,0 +1,16 @@
+require "jsonapi/serializer"
+
+module EasyML
+  class PredictionSerializer
+    include JSONAPI::Serializer
+
+    attribute :prediction do |object|
+      object.prediction_value.symbolize_keys.dig(:value)
+    end
+
+    attributes :id,
+               :prediction_type,
+               :raw_input,
+               :normalized_input
+  end
+end

data/config/routes.rb

@@ -1,4 +1,5 @@
 require "zhong/web"
+require "resque/server"
 
 EasyML::Engine.routes.draw do
   root to: "models#index"

data/lib/easy_ml/predict.rb

@@ -29,8 +29,6 @@ module EasyML
         raw_input: raw_input,
         normalized_input: df.to_hashes&.first,
       )
-
-      preds
     end
 
     def self.train(model_name, tuner: nil, evaluator: nil)

data/lib/easy_ml/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: easy_ml
 version: !ruby/object:Gem::Version
-  version: 0.2.0.pre.rc38
+  version: 0.2.0.pre.rc39
 platform: ruby
 authors:
 - Brett Shollenberger
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-01-22 00:00:00.000000000 Z
+date: 2025-01-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -594,6 +594,7 @@ files:
 - app/serializers/easy_ml/datasource_serializer.rb
 - app/serializers/easy_ml/feature_serializer.rb
 - app/serializers/easy_ml/model_serializer.rb
+- app/serializers/easy_ml/prediction_serializer.rb
 - app/serializers/easy_ml/retraining_job_serializer.rb
 - app/serializers/easy_ml/retraining_run_serializer.rb
 - app/serializers/easy_ml/settings_serializer.rb