Skip to content
Snippets Groups Projects
Commit c25c1b17 authored by Jashan's avatar Jashan
Browse files

Add Skeleton Code

parent 5059977d
No related branches found
No related tags found
No related merge requests found
Pipeline #1411 passed
.DS_Store 0 → 100644
File added
import ModelCreation
class DecisionTreeCreation(ModelCreation):
def __init__(self):
super().__init__()
import ModelCreation
class LogRegressionCreation(ModelCreation):
def __init__(self):
super().__init__()
\ No newline at end of file
import pandas as pd
from sklearn.model_selection import train_test_split
import numpy as np
class ModelCreation:
def __init__(self, data, filename, averagePrecision):
self.data = data
self.filename = filename
self.averagePrecision = averagePrecision
def construct(self, data):
return
def fit(self):
return
def save(self, filename):
return
def predictProbability(self):
return
def ROC(self):
return
def AUC(self):
return
def precisionCurve(self):
return
def averagePrecision(self):
return
\ No newline at end of file
import ModelCreation
class RandomForestCreation(ModelCreation):
def __init__(self):
super().__init__()
\ No newline at end of file
import ModelCreation
class XGBoostCreation(ModelCreation):
def __init__(self):
super().__init__()
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment