Learn R Programming

⚠️There's a newer version (1.3.4.1) of this package.Take me there.

bartMachine (version 1.2.6)

Bayesian Additive Regression Trees

Description

An advanced implementation of Bayesian Additive Regression Trees with expanded features for data analysis and visualization.

Copy Link

Version

Install

install.packages('bartMachine')

Monthly Downloads

1,876

Version

1.2.6

License

GPL-3

Maintainer

Adam Kapelner

Last Published

December 17th, 2020

Functions in bartMachine (1.2.6)

bart_predict_for_test_data

Predict for Test Data with Known Outcomes
benchmark_datasets

benchmark_datasets
linearity_test

Test of Linearity
dummify_data

Dummify Design Matrix
bartMachine

Build a BART Model
destroy_bart_machine

Destroy BART Model (deprecated --- do not use!)
node_prediction_training_data_indices

Gets node predictions indices of the training data for new data.
var_selection_by_permute_cv

Perform Variable Selection Using Cross-validation Procedure
investigate_var_importance

Explore Variable Inclusion Proportions in BART Model
bart_machine_get_posterior

Get Full Posterior Distribution
bart_machine_num_cores

Get Number of Cores Used by BART
bartMachineArr

Create an array of BART models for the same data.
bartMachineCV

Build BART-CV
k_fold_cv

Estimate Out-of-sample Error with K-fold Cross validation
get_var_props_over_chain

Get the Variable Inclusion Proportions
plot_y_vs_yhat

Plot the fitted Versus Actual Response
interaction_investigator

Explore Pairwise Interactions in BART Model
check_bart_error_assumptions

Check BART Error Assumptions
extract_raw_node_data

Gets Raw Node data
predict.bartMachine

Make a prediction on data using a BART object
get_projection_weights

Gets Training Sample Projection / Weights
rmse_by_num_trees

Assess the Out-of-sample RMSE by Number of Trees
pd_plot

Partial Dependence Plot
cov_importance_test

Importance Test for Covariate(s) of Interest
plot_convergence_diagnostics

Plot Convergence Diagnostics
set_bart_machine_num_cores

Set the Number of Cores for BART
calc_credible_intervals

Calculate Credible Intervals
calc_prediction_intervals

Calculate Prediction Intervals
get_sigsqs

Get Posterior Error Variance Estimates
predict_bartMachineArr

Make a prediction on data using a BART array object
get_var_counts_over_chain

Get the Variable Inclusion Counts
print.bartMachine

Summarizes information about a bartMachine object.
var_selection_by_permute

Perform Variable Selection using Three Threshold-based Procedures
summary.bartMachine

Summarizes information about a bartMachine object.
automobile

Data concerning automobile prices.