Added ability to store CNN models
All checks were successful
continuous-integration/drone/push Build is passing
All checks were successful
continuous-integration/drone/push Build is passing
This commit is contained in:
16
model_service/model/svmdetails.py
Normal file
16
model_service/model/svmdetails.py
Normal file
@ -0,0 +1,16 @@
|
||||
#!/usr/bin/env python3
|
||||
from .db import db
|
||||
from sqlalchemy.dialects.postgresql import UUID
|
||||
|
||||
|
||||
class SVMDetails(db.Model):
|
||||
aimodel_id = db.Column(UUID(as_uuid=True), db.ForeignKey("ai_model.id"), nullable=False, primary_key=True)
|
||||
aimodel = db.relationship("AIModel", backref=db.backref("details", lazy=True, cascade="save-update, merge, delete, delete-orphan"))
|
||||
|
||||
# details
|
||||
mid_window = db.Column(db.Float)
|
||||
mid_step = db.Column(db.Float)
|
||||
short_window = db.Column(db.Float)
|
||||
short_step = db.Column(db.Float)
|
||||
compute_beat = db.Column(db.Boolean)
|
||||
|
Reference in New Issue
Block a user