├── .Rbuildignore ├── .github ├── .gitignore ├── CONTRIBUTING.md └── workflows │ └── pr-commands.yaml ├── .gitignore ├── CODE_OF_CONDUCT.md ├── DESCRIPTION ├── LICENSE.md ├── NAMESPACE ├── R ├── bert_classification-constructor.R ├── bert_classification-fit.R ├── bert_classification-predict.R ├── bert_linear.R ├── bert_regression-constructor.R ├── bert_regression-fit.R ├── bert_regression-predict.R ├── imports.R ├── parsnip.R ├── serialization.R ├── tidy_output.R ├── tidybert-package.R └── validation.R ├── README.Rmd ├── README.md ├── man ├── bert.Rd ├── bert_classification.Rd ├── bert_regression.Rd ├── bert_type.Rd ├── dot-bert_classification_bridge.Rd ├── dot-bert_classification_impl.Rd ├── dot-bert_regression_bridge.Rd ├── dot-bert_regression_impl.Rd ├── dot-check_luz_model_serialization.Rd ├── dot-check_predictors_are_character.Rd ├── dot-finalize_attention.Rd ├── dot-finalize_embeddings.Rd ├── dot-get_bert_classification_predict_function.Rd ├── dot-is_null_external_pointer.Rd ├── dot-make_bert.Rd ├── dot-make_index_column.Rd ├── dot-make_token_map.Rd ├── dot-mold_valid_formula.Rd ├── dot-mold_valid_xy.Rd ├── dot-new_bert_classification.Rd ├── dot-new_bert_regression.Rd ├── dot-predict_bert_classification_bridge.Rd ├── dot-predict_bert_classification_class.Rd ├── dot-predict_bert_classification_prob.Rd ├── dot-predict_bert_classification_shared.Rd ├── dot-predict_bert_regression.Rd ├── dot-predict_bert_regression_bridge.Rd ├── dot-process_attention_result.Rd ├── dot-process_embeddings_result.Rd ├── dot-serialize_luz_model.Rd ├── dot-unserialize_luz_model.Rd ├── dot-validate_predictor_count.Rd ├── dot-validate_predictors_are_character.Rd ├── model_bert_linear.Rd ├── n_tokens.Rd ├── pipe.Rd ├── predict.bert_classification.Rd ├── predict.bert_regression.Rd ├── reexports.Rd ├── tidy_bert_output.Rd ├── tidybert-package.Rd └── tidybert_engine.Rd ├── tests ├── testthat.R └── testthat │ ├── _snaps │ ├── bert_classification-fit.md │ ├── bert_classification-predict.md │ ├── bert_linear.md │ ├── bert_regression-fit.md │ ├── parsnip.md │ ├── serialization.md │ ├── tidy_output.md │ └── validation.md │ ├── test-bert_classification-fit.R │ ├── test-bert_classification-predict.R │ ├── test-bert_linear.R │ ├── test-bert_regression-fit.R │ ├── test-bert_regression-predict.R │ ├── test-parsnip.R │ ├── test-serialization.R │ ├── test-tidy_output.R │ └── test-validation.R ├── tidybert.Rproj └── vignettes ├── .gitignore ├── basic_usage.Rmd ├── bert_classification.Rmd └── tidymodels.Rmd /.Rbuildignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/.Rbuildignore -------------------------------------------------------------------------------- /.github/.gitignore: -------------------------------------------------------------------------------- 1 | *.html 2 | -------------------------------------------------------------------------------- /.github/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/.github/CONTRIBUTING.md -------------------------------------------------------------------------------- /.github/workflows/pr-commands.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/.github/workflows/pr-commands.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/.gitignore -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /DESCRIPTION: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/DESCRIPTION -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/LICENSE.md -------------------------------------------------------------------------------- /NAMESPACE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/NAMESPACE -------------------------------------------------------------------------------- /R/bert_classification-constructor.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/R/bert_classification-constructor.R -------------------------------------------------------------------------------- /R/bert_classification-fit.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/R/bert_classification-fit.R -------------------------------------------------------------------------------- /R/bert_classification-predict.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/R/bert_classification-predict.R -------------------------------------------------------------------------------- /R/bert_linear.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/R/bert_linear.R -------------------------------------------------------------------------------- /R/bert_regression-constructor.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/R/bert_regression-constructor.R -------------------------------------------------------------------------------- /R/bert_regression-fit.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/R/bert_regression-fit.R -------------------------------------------------------------------------------- /R/bert_regression-predict.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/R/bert_regression-predict.R -------------------------------------------------------------------------------- /R/imports.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/R/imports.R -------------------------------------------------------------------------------- /R/parsnip.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/R/parsnip.R -------------------------------------------------------------------------------- /R/serialization.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/R/serialization.R -------------------------------------------------------------------------------- /R/tidy_output.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/R/tidy_output.R -------------------------------------------------------------------------------- /R/tidybert-package.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/R/tidybert-package.R -------------------------------------------------------------------------------- /R/validation.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/R/validation.R -------------------------------------------------------------------------------- /README.Rmd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/README.Rmd -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/README.md -------------------------------------------------------------------------------- /man/bert.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/bert.Rd -------------------------------------------------------------------------------- /man/bert_classification.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/bert_classification.Rd -------------------------------------------------------------------------------- /man/bert_regression.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/bert_regression.Rd -------------------------------------------------------------------------------- /man/bert_type.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/bert_type.Rd -------------------------------------------------------------------------------- /man/dot-bert_classification_bridge.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-bert_classification_bridge.Rd -------------------------------------------------------------------------------- /man/dot-bert_classification_impl.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-bert_classification_impl.Rd -------------------------------------------------------------------------------- /man/dot-bert_regression_bridge.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-bert_regression_bridge.Rd -------------------------------------------------------------------------------- /man/dot-bert_regression_impl.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-bert_regression_impl.Rd -------------------------------------------------------------------------------- /man/dot-check_luz_model_serialization.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-check_luz_model_serialization.Rd -------------------------------------------------------------------------------- /man/dot-check_predictors_are_character.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-check_predictors_are_character.Rd -------------------------------------------------------------------------------- /man/dot-finalize_attention.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-finalize_attention.Rd -------------------------------------------------------------------------------- /man/dot-finalize_embeddings.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-finalize_embeddings.Rd -------------------------------------------------------------------------------- /man/dot-get_bert_classification_predict_function.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-get_bert_classification_predict_function.Rd -------------------------------------------------------------------------------- /man/dot-is_null_external_pointer.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-is_null_external_pointer.Rd -------------------------------------------------------------------------------- /man/dot-make_bert.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-make_bert.Rd -------------------------------------------------------------------------------- /man/dot-make_index_column.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-make_index_column.Rd -------------------------------------------------------------------------------- /man/dot-make_token_map.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-make_token_map.Rd -------------------------------------------------------------------------------- /man/dot-mold_valid_formula.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-mold_valid_formula.Rd -------------------------------------------------------------------------------- /man/dot-mold_valid_xy.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-mold_valid_xy.Rd -------------------------------------------------------------------------------- /man/dot-new_bert_classification.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-new_bert_classification.Rd -------------------------------------------------------------------------------- /man/dot-new_bert_regression.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-new_bert_regression.Rd -------------------------------------------------------------------------------- /man/dot-predict_bert_classification_bridge.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-predict_bert_classification_bridge.Rd -------------------------------------------------------------------------------- /man/dot-predict_bert_classification_class.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-predict_bert_classification_class.Rd -------------------------------------------------------------------------------- /man/dot-predict_bert_classification_prob.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-predict_bert_classification_prob.Rd -------------------------------------------------------------------------------- /man/dot-predict_bert_classification_shared.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-predict_bert_classification_shared.Rd -------------------------------------------------------------------------------- /man/dot-predict_bert_regression.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-predict_bert_regression.Rd -------------------------------------------------------------------------------- /man/dot-predict_bert_regression_bridge.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-predict_bert_regression_bridge.Rd -------------------------------------------------------------------------------- /man/dot-process_attention_result.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-process_attention_result.Rd -------------------------------------------------------------------------------- /man/dot-process_embeddings_result.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-process_embeddings_result.Rd -------------------------------------------------------------------------------- /man/dot-serialize_luz_model.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-serialize_luz_model.Rd -------------------------------------------------------------------------------- /man/dot-unserialize_luz_model.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-unserialize_luz_model.Rd -------------------------------------------------------------------------------- /man/dot-validate_predictor_count.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-validate_predictor_count.Rd -------------------------------------------------------------------------------- /man/dot-validate_predictors_are_character.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/dot-validate_predictors_are_character.Rd -------------------------------------------------------------------------------- /man/model_bert_linear.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/model_bert_linear.Rd -------------------------------------------------------------------------------- /man/n_tokens.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/n_tokens.Rd -------------------------------------------------------------------------------- /man/pipe.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/pipe.Rd -------------------------------------------------------------------------------- /man/predict.bert_classification.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/predict.bert_classification.Rd -------------------------------------------------------------------------------- /man/predict.bert_regression.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/predict.bert_regression.Rd -------------------------------------------------------------------------------- /man/reexports.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/reexports.Rd -------------------------------------------------------------------------------- /man/tidy_bert_output.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/tidy_bert_output.Rd -------------------------------------------------------------------------------- /man/tidybert-package.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/tidybert-package.Rd -------------------------------------------------------------------------------- /man/tidybert_engine.Rd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/man/tidybert_engine.Rd -------------------------------------------------------------------------------- /tests/testthat.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tests/testthat.R -------------------------------------------------------------------------------- /tests/testthat/_snaps/bert_classification-fit.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tests/testthat/_snaps/bert_classification-fit.md -------------------------------------------------------------------------------- /tests/testthat/_snaps/bert_classification-predict.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tests/testthat/_snaps/bert_classification-predict.md -------------------------------------------------------------------------------- /tests/testthat/_snaps/bert_linear.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tests/testthat/_snaps/bert_linear.md -------------------------------------------------------------------------------- /tests/testthat/_snaps/bert_regression-fit.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tests/testthat/_snaps/bert_regression-fit.md -------------------------------------------------------------------------------- /tests/testthat/_snaps/parsnip.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tests/testthat/_snaps/parsnip.md -------------------------------------------------------------------------------- /tests/testthat/_snaps/serialization.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tests/testthat/_snaps/serialization.md -------------------------------------------------------------------------------- /tests/testthat/_snaps/tidy_output.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tests/testthat/_snaps/tidy_output.md -------------------------------------------------------------------------------- /tests/testthat/_snaps/validation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tests/testthat/_snaps/validation.md -------------------------------------------------------------------------------- /tests/testthat/test-bert_classification-fit.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tests/testthat/test-bert_classification-fit.R -------------------------------------------------------------------------------- /tests/testthat/test-bert_classification-predict.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tests/testthat/test-bert_classification-predict.R -------------------------------------------------------------------------------- /tests/testthat/test-bert_linear.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tests/testthat/test-bert_linear.R -------------------------------------------------------------------------------- /tests/testthat/test-bert_regression-fit.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tests/testthat/test-bert_regression-fit.R -------------------------------------------------------------------------------- /tests/testthat/test-bert_regression-predict.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tests/testthat/test-bert_regression-predict.R -------------------------------------------------------------------------------- /tests/testthat/test-parsnip.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tests/testthat/test-parsnip.R -------------------------------------------------------------------------------- /tests/testthat/test-serialization.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tests/testthat/test-serialization.R -------------------------------------------------------------------------------- /tests/testthat/test-tidy_output.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tests/testthat/test-tidy_output.R -------------------------------------------------------------------------------- /tests/testthat/test-validation.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tests/testthat/test-validation.R -------------------------------------------------------------------------------- /tidybert.Rproj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/tidybert.Rproj -------------------------------------------------------------------------------- /vignettes/.gitignore: -------------------------------------------------------------------------------- 1 | *.html 2 | *.R 3 | -------------------------------------------------------------------------------- /vignettes/basic_usage.Rmd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/vignettes/basic_usage.Rmd -------------------------------------------------------------------------------- /vignettes/bert_classification.Rmd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/vignettes/bert_classification.Rmd -------------------------------------------------------------------------------- /vignettes/tidymodels.Rmd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/macmillancontentscience/tidybert/HEAD/vignettes/tidymodels.Rmd --------------------------------------------------------------------------------