Esempio n. 1
0
class DataSourceModel(BaseModel):
    _id = UUIDType(required=True)

    file_name = StringType(required=True)
    file_path = StringType(required=True)
    file_sep = StringType()
    pickle_file_path = StringType()
    content_type = StringType(required=True)
    file_name_md5 = StringType()
    file_size = StringType()
    uploaded_at = DateTimeType(default=datetime.datetime.now())

    samples_count = IntType(default=0)
    predictors_count = IntType(default=0)
    nominal_predictors_count = IntType(default=0)
    numeric_predictors_count = IntType(default=0)
    class_count = IntType(default=0)
    missing_values = FloatType(default=.0)

    status = StringType(default='check')

    predictor_values = ListType(StringType())
    predictor_details = ListType(DictType(BaseType))

    predictor_ids = ListType(StringType())
    predictor_target_name = StringType()

    MONGO_COLLECTION = 'datasources'
Esempio n. 2
0
class WorkflowModel(BaseModel):
    _id = UUIDType(required=True)

    name = StringType()
    created_at = DateTimeType(default=datetime.datetime.now())
    processing_started_at = DateTimeType()
    processing_ended_at = DateTimeType()

    training_data = DictType(BaseType, {})
    preprocessing = ListType(StringType, default=list())
    model_processing = DictType(BaseType, default={'type': 'supervised'})
    validation = DictType(BaseType, default={'type': 'fold', 'value': 3})
    trained_models = DictType(BaseType, {})
    fi_booster = StringType()

    validation_details = DictType(BaseType, default={})

    state = StringType(default='new')

    MONGO_COLLECTION = 'workflows'