sys.path.append(EnsemblePath)
from RandomForest import RandomForest

# %%

train = 'D:\\School\\Spring 2021\\CS 6350\\Homework\\HW2\\bank\\train.csv'
test = 'D:\\School\\Spring 2021\\CS 6350\\Homework\\HW2\\bank\\test.csv'

df = pd.read_csv(train, header=None)
TreeHelper.ProcessDataFrame(df)


forest = RandomForest(df, 2)


forest.BuildNTrees(1)



# %% 
test_df = pd.read_csv(test, header=None)

TreeHelper.ProcessDataFrame(test_df)
# %%

print(forest.Predict(test_df.iloc[102,:]))

# %% Varying Number of Trees

train = 'D:\\School\\Spring 2021\\CS 6350\\Homework\\HW2\\bank\\train.csv'
test = 'D:\\School\\Spring 2021\\CS 6350\\Homework\\HW2\\bank\\test.csv'