Adisri99 commited on
Commit
b6180aa
·
verified ·
1 Parent(s): e80c00c

Update backend/app/models/experiment.py

Browse files
Files changed (1) hide show
  1. backend/app/models/experiment.py +8 -16
backend/app/models/experiment.py CHANGED
@@ -1,24 +1,16 @@
1
- from sqlalchemy import Column, Float, ForeignKey, Integer, String, Text
2
  from backend.app.db import Base
3
 
 
4
  class Experiment(Base):
5
  __tablename__ = "experiments"
 
6
  id = Column(String, primary_key=True, index=True)
7
- dataset_id = Column(String, ForeignKey("datasets.id"), nullable=False, index=True)
8
- name = Column(String, nullable=False)
9
  algorithm = Column(String, nullable=False)
10
  status = Column(String, nullable=False)
11
  config_json = Column(Text, nullable=False)
12
- metrics_json = Column(Text, nullable=False)
13
- summaries_json = Column(Text, nullable=False)
14
- output_csv_path = Column(String, nullable=False)
15
-
16
- class ClusterResult(Base):
17
- __tablename__ = "cluster_results"
18
- id = Column(Integer, primary_key=True, autoincrement=True)
19
- experiment_id = Column(String, ForeignKey("experiments.id"), nullable=False, index=True)
20
- row_index = Column(Integer, nullable=False)
21
- cluster_label = Column(Integer, nullable=False)
22
- is_outlier = Column(Integer, nullable=False, default=0)
23
- x = Column(Float, nullable=False, default=0.0)
24
- y = Column(Float, nullable=False, default=0.0)
 
1
+ from sqlalchemy import Column, Integer, String, Text
2
  from backend.app.db import Base
3
 
4
+
5
  class Experiment(Base):
6
  __tablename__ = "experiments"
7
+
8
  id = Column(String, primary_key=True, index=True)
9
+ dataset_id = Column(String, nullable=False, index=True)
 
10
  algorithm = Column(String, nullable=False)
11
  status = Column(String, nullable=False)
12
  config_json = Column(Text, nullable=False)
13
+ metrics_json = Column(Text, nullable=True)
14
+ summary_json = Column(Text, nullable=True)
15
+ runtime_ms = Column(Integer, nullable=True)
16
+ error_message = Column(Text, nullable=True)