Train-your-own-Model/Model_Registry.py at main · BEGINNERUSER-git/Train-your-own-Model · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
from Model import(
    KmeansSklearn,
    KmeansScratch,
    BinaryClassificationScratch,
    BinaryClassificationSklearn,
    DecisionTreeScratch,
    DecisionTreeClassifierSklearn,
    AnomalyDetection,
    LinearRegressionScratch,
    LinearRegressionSklearn,
    MulticlassScratch,
    MulticlassShallowNN,
    RandomForestClassifierSklearn,
    XGBoostClassifierSklearn

)

MODEL_REGISTRY={
    "Clustering":{
    "Kmeans":{
        "sklearn":KmeansSklearn,
        "scratch":KmeansScratch
    }
    },
    "Classification":{
    "Binary_Classification":{
        "sklearn":BinaryClassificationSklearn,
        "scratch":BinaryClassificationScratch
    },
    "Decision_Tree":{
        "sklearn":DecisionTreeClassifierSklearn,
        "scratch":DecisionTreeScratch
    },

    "Random_Forest":{
        "sklearn":RandomForestClassifierSklearn
    },
    "XGBoost":{
        "sklearn":XGBoostClassifierSklearn
    },},

    "Anomaly_Detection":{
    "Gaussian Anomaly":{
        "scratch":AnomalyDetection
    },
},
    "Regression":{
    "Linear_Regression":{
        "sklearn":    LinearRegressionSklearn,
        "scratch":LinearRegressionScratch
    }},
    "Multiclass_Classification":{
    "softmax_NN":{
        "shallow_NN":MulticlassShallowNN,
        "scratch":MulticlassScratch
    },}

}