Skip to content

Instantly share code, notes, and snippets.

@KenoLeon
Last active July 17, 2020 19:51
Show Gist options
  • Select an option

  • Save KenoLeon/d92837d821d6ee5c5a678a0f63c2b5d6 to your computer and use it in GitHub Desktop.

Select an option

Save KenoLeon/d92837d821d6ee5c5a678a0f63c2b5d6 to your computer and use it in GitHub Desktop.
Load Numerai dataset.
import pandas as pd
# CONSTANTS:
ROUND = '220'
TOURNAMENT_NAME = "kazutsugi"
TARGET_NAME = f"target_{TOURNAMENT_NAME}"
PREDICTION_NAME = f"prediction_{TOURNAMENT_NAME}"
# LOAD DATASETS:
training_data = pd.read_csv("Numerai/Datasets/numerai_dataset_"+ROUND+"/numerai_training_data.csv").set_index("id")
tournament_data = pd.read_csv("Numerai/Datasets/numerai_dataset_"+ROUND+"/numerai_tournament_data.csv").set_index("id")
# FEATURES
feature_names = [
f for f in training_data.columns if f.startswith("feature")
]
print(f"Loaded {len(feature_names)} features columns")
train_Features = training_data[feature_names]
train_Target = training_data[TARGET_NAME]
print(':::::: train_Features ::::::', '\n', train_Features)
print(':::::: train_Target ::::::', '\n', train_Target)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment