FinCoach / main.py
WNSachio's picture
feat: add retraining function to retrain model from new data
00b69d2
raw
history blame contribute delete
385 Bytes
from pipeline.extract import fetch_close_data
from pipeline.transform import preprocess_data
from pipeline.load import save_to_csv
from pipeline import retrain
def main():
df_raw = fetch_close_data()
df_clean = preprocess_data(df_raw)
csv_path = "./stocks.csv"
save_to_csv(df_clean, csv_path)
# Retrain
retrain.main()
if __name__ == "__main__":
main()