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

Add Skeleton Code

parent 5059977d
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 to comment