def train_rf_squad():
dataset = import_data(None, True)
x_train, y, feature_names = feature_engineering(dataset, 'squad-fpp', True)
model = rf_model_squad(x_train, y)
print("Model training completed successfully")
print(model)
def model_training():
dataset = import_data(None, True)
x_train, y = feature_engineering(dataset, True)
model = rf_model_solo(x_train, y)
print(model)
def ramdon_forest_duo(file):
dataset = import_data(file, False)
df, y, feature_names = feature_engineering(dataset, 'duo-fpp', False)
ans = rf_duo(df)
return ans
def ramdon_forest_solo(file):
dataset = import_data(file, False)
df, _ = feature_engineering(dataset, False)
ans = random_forest(df)
return ans