File List

Here is a list of all files with brief descriptions:

[detail level 123456789]

▼ include
► cuml
► cluster
dbscan.hpp
dbscan_api.h
hdbscan.hpp
kmeans.hpp
kmeans_mg.hpp
linkage.hpp
spectral.hpp
► common
callback.hpp
callbackSink.hpp
log_levels.hpp
logger.hpp
pinned_host_vector.hpp
utils.hpp
► datasets
make_arima.hpp
make_blobs.hpp
make_regression.hpp
► decomposition
params.hpp
pca.hpp
pca_mg.hpp
sign_flip_mg.hpp
tsvd.hpp
tsvd_mg.hpp
► ensemble
randomforest.hpp
treelite_defs.hpp
► experimental
► fil
► detail
► device_initialization
cpu.hpp
gpu.cuh
gpu.hpp
► infer
cpu.hpp
gpu.cuh
gpu.hpp
► infer_kernel
cpu.hpp
gpu.cuh
shared_memory_buffer.cuh
► raft_proto
► detail
► copy
cpu.hpp
gpu.hpp
► cuda_check
base.hpp
gpu.hpp
► device_id
base.hpp
cpu.hpp
gpu.hpp
► device_setter
base.hpp
gpu.hpp
► host_only_throw
base.hpp
cpu.hpp
► non_owning_buffer
base.hpp
► owning_buffer
base.hpp
cpu.hpp
gpu.hpp
const_agnostic.hpp
copy.hpp
host_only_throw.hpp
non_owning_buffer.hpp
owning_buffer.hpp
buffer.hpp
ceildiv.hpp
cuda_check.hpp
cuda_stream.hpp
device_id.hpp
device_setter.hpp
device_type.hpp
exceptions.hpp
gpu_support.hpp
handle.hpp
padding.hpp
► specializations
device_initialization_macros.hpp
forest_macros.hpp
infer_macros.hpp
bitset.hpp
cpu_introspection.hpp
decision_forest_builder.hpp
device_initialization.hpp
evaluate_tree.hpp
forest.hpp
gpu_introspection.hpp
index_type.hpp
infer.hpp
node.hpp
postprocessor.hpp
specialization_types.hpp
constants.hpp
decision_forest.hpp
exceptions.hpp
forest_model.hpp
infer_kind.hpp
postproc_ops.hpp
tree_layout.hpp
treelite_importer.hpp
► explainer
kernel_shap.hpp
permutation_shap.hpp
tree_shap.hpp
► fil
fil.h
fnv_hash.h
multi_sum.cuh
► genetic
common.h
genetic.h
node.h
program.h
► linear_model
glm.hpp
glm_api.h
ols_mg.hpp
preprocess_mg.hpp
qn.h
qn_mg.hpp
ridge_mg.hpp
► manifold
common.hpp
tsne.h
umap.hpp
umapparams.h
► matrix
kernelparams.h
► metrics
metrics.hpp
► neighbors
knn.hpp
knn_api.h
knn_mg.hpp
knn_sparse.hpp
► random_projection
rproj_c.h
► solvers
cd_mg.hpp
lars.hpp
params.hpp
solver.hpp
► svm
linear.hpp
svc.hpp
svm_api.h
svm_model.h
svm_parameter.h
svr.hpp
► tree
algo_helper.h
decisiontree.hpp
flatnode.h
► tsa
arima_common.h
auto_arima.h
batched_arima.hpp
batched_kalman.hpp
holtwinters.h
holtwinters_api.h
holtwinters_params.h
stationarity.h
cuml_api.h
▼ src
► arima
batched_arima.cu
batched_kalman.cu
► common
cuml_api.cpp
cumlHandle.cpp
cumlHandle.hpp
logger.cpp
nvtx.hpp
► datasets
make_arima.cu
make_blobs.cu
make_regression.cu
► dbscan
► adjgraph
algo.cuh
pack.h
runner.cuh
► corepoints
compute.cuh
exchange.cuh
► mergelabels
runner.cuh
tree_reduction.cuh
► vertexdeg
algo.cuh
pack.h
precomputed.cuh
runner.cuh
dbscan.cu
dbscan.cuh
dbscan_api.cpp
runner.cuh
► decisiontree
► batched-levelalgo
► kernels
builder_kernels.cuh
builder_kernels_impl.cuh
entropy-double.cu
entropy-float.cu
gamma-double.cu
gamma-float.cu
gini-double.cu
gini-float.cu
inverse_gaussian-double.cu
inverse_gaussian-float.cu
mse-double.cu
mse-float.cu
poisson-double.cu
poisson-float.cu
quantiles.cu
bins.cuh
builder.cuh
dataset.h
objectives.cuh
quantiles.cuh
quantiles.h
split.cuh
decisiontree.cu
decisiontree.cuh
treelite_util.h
► experimental
► fil
infer0.cpp
infer0.cu
infer1.cpp
infer1.cu
infer2.cpp
infer2.cu
infer3.cpp
infer3.cu
infer4.cpp
infer4.cu
infer5.cpp
infer5.cu
infer6.cpp
infer6.cu
infer7.cpp
infer7.cu
► explainer
kernel_shap.cu
permutation_shap.cu
tree_shap.cu
► fil
common.cuh
fil.cu
infer.cu
internal.cuh
treelite_import.cu
► genetic
constants.h
fitness.cuh
genetic.cu
node.cu
node.cuh
program.cu
reg_stack.cuh
► glm
► qn
► mg
glm_base_mg.cuh
qn_mg.cuh
standardization.cuh
► simple_mat
base.hpp
dense.hpp
sparse.hpp
glm_base.cuh
glm_linear.cuh
glm_logistic.cuh
glm_regularizer.cuh
glm_softmax.cuh
glm_svm.cuh
qn.cuh
qn_linesearch.cuh
qn_solvers.cuh
qn_util.cuh
simple_mat.cuh
glm.cu
glm_api.cpp
ols.cuh
ols_mg.cu
preprocess.cuh
preprocess_mg.cu
qn_mg.cu
ridge.cuh
ridge_mg.cu
► hdbscan
► detail
► kernels
condense.cuh
membership.cuh
predict.cuh
select.cuh
soft_clustering.cuh
stabilities.cuh
condense.cuh
extract.cuh
membership.cuh
predict.cuh
reachability.cuh
select.cuh
soft_clustering.cuh
stabilities.cuh
utils.h
condensed_hierarchy.cu
hdbscan.cu
prediction_data.cu
runner.h
► hierarchy
linkage.cu
► holtwinters
► internal
hw_decompose.cuh
hw_eval.cuh
hw_forecast.cuh
hw_optim.cuh
hw_utils.cuh
holtwinters.cu
holtwinters_api.cpp
runner.cuh
► kmeans
kmeans_fit_predict.cu
kmeans_mg.cu
kmeans_predict.cu
kmeans_transform.cu
► knn
knn.cu
knn_api.cpp
knn_classify_mg.cu
knn_mg.cu
knn_opg_common.cuh
knn_regress_mg.cu
knn_sparse.cu
► metrics
accuracy_score.cu
adjusted_rand_index.cu
completeness_score.cu
entropy.cu
homogeneity_score.cu
kl_divergence.cu
mutual_info_score.cu
pairwise_distance.cu
r2_score.cu
rand_index.cu
silhouette_score.cu
silhouette_score_batched_double.cu
silhouette_score_batched_float.cu
trustworthiness.cu
v_measure.cu
► pca
pca.cu
pca.cuh
pca_mg.cu
sign_flip_mg.cu
► random_projection
rproj.cu
rproj.cuh
rproj_utils.cuh
► randomforest
randomforest.cu
randomforest.cuh
► solver
cd.cuh
cd_mg.cu
lars.cu
lars_impl.cuh
learning_rate.h
sgd.cuh
shuffle.h
solver.cu
► spectral
spectral.cu
► svm
kernelcache.cuh
linear.cu
results.cuh
smo_sets.cuh
smoblocksolve.cuh
smosolver.cuh
smosolver.h
sparse_util.cuh
svc.cu
svc_impl.cuh
svm_api.cpp
svr.cu
svr_impl.cuh
workingset.cuh
workingset.h
ws_util.cuh
► tsa
auto_arima.cu
auto_arima.cuh
stationarity.cu
► tsne
► cannylab
bh.cu
barnes_hut_kernels.cuh
barnes_hut_tsne.cuh
distances.cuh
exact_kernels.cuh
exact_tsne.cuh
fft_kernels.cuh
fft_tsne.cuh
tsne.cu
tsne_runner.cuh
utils.cuh
► tsvd
tsvd.cu
tsvd.cuh
tsvd_mg.cu
► umap
► fuzzy_simpl_set
naive.cuh
runner.cuh
► init_embed
random_algo.cuh
runner.cuh
spectral_algo.cuh
► knn_graph
algo.cuh
runner.cuh
► simpl_set_embed
algo.cuh
optimize_batch_kernel.cuh
runner.cuh
optimize.cuh
runner.cuh
supervised.cuh
umap.cu
ml_cuda_utils.h
ml_mg_utils.cuh
▼ src_prims
► common
device_loads_stores.cuh
device_utils.cuh
fast_int_div.cuh
grid_sync.cuh
iota.cuh
Timer.h
► datasets
boston.h
breast_cancer.h
diabetes.h
digits.h
► functions
hinge.cuh
linearReg.cuh
log.cuh
logisticReg.cuh
penalty.cuh
sigmoid.cuh
sign.cuh
softThres.cuh
► linalg
► batched
gemv.cuh
make_symm.cuh
matrix.cuh
block.cuh
custom_accum.h
eltwise2d.cuh
init.h
► matrix
reverse.cuh
► random
make_arima.cuh
► selection
knn.cuh
kselection.cuh
► sparse
► batched
csr.cuh
► timeSeries
arima_helpers.cuh
fillna.cuh
jones_transform.cuh
stationarity.cuh
cufft_utils.h
decoupled_lookback.cuh