ufc-predictor / config.py
zjpiazza's picture
Updated file structure and added predictor.py
cf2059c
raw
history blame contribute delete
572 Bytes
from pathlib import Path
# Base paths
ROOT_DIR = Path(__file__).parent
DATA_DIR = ROOT_DIR / "data"
# Model paths
MODEL_PATH = DATA_DIR / "model" / "model.h5"
# Data paths
FIGHT_STATS_PATH = DATA_DIR / "preprocessing" / "fight_with_stats_precomp.csv"
FIGHTER_STATS_PATH = DATA_DIR / "preprocessing" / "fighter_total_stats.csv"
FIGHTER_DETAILS_PATH = DATA_DIR / "preprocessing" / "fighter_details.csv"
MODEL_DATA_PATH = DATA_DIR / "model" / "df_model.csv"
# TensorFlow config
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'