Example #1
0
def test_dump():

    predictions = [Prediction(None, None, None, None, None)]
    algo = AlgoBase()
    trainset = Trainset(*[None] * 9)

    with tempfile.NamedTemporaryFile() as tmp_file:
        dump(tmp_file.name, predictions, trainset, algo)
 def __init__(self):
     self.train_x = pd.DataFrame()
     self.train_y = pd.DataFrame()
     self.val_x = pd.DataFrame()
     self.val_y = pd.DataFrame()
     self.predicted_y = list()
     self.y = np.zeros(100)
     self.reg = LinearRegression()
     self.knn = AlgoBase()