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()