File size: 1,808 Bytes
51379b3 c352e09 51379b3 c352e09 51379b3 0a0b812 c352e09 51379b3 50cf060 328514c 0a0b812 bc9f87f cad4512 43d4a4a bc9f87f fe8dbb7 bc9f87f ab4ccef fe8dbb7 bc9f87f 0a0b812 fe8dbb7 bc9f87f fe8dbb7 675e4b0 43d4a4a bc9f87f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
---
library_name: sklearn
tags:
- tabular-regression
- materials property prediction
- baseline-trainer
---
**Model Description**
The magnet Curie temperature (Tc [K]) predictor model has been trained using a supervised learning approach on a specific set of magnet classes having 14:2:1 phases.
The dataset to train the Tc prediction model is a distinct literature source.
Further, the Tc values for various 14:2:1 magnet phases at room temperature are considered for dataset creation.
It predicts the Tc value using the chemical composition as a feature.
E.g: To predict the Tc value Nd2Fe14B1 magnet composition, the features are Nd=2, Fe=14, and B=1.
**Application & Limitations**
Input feature as the chemical composition of the test sample should match the sequence of the features described in the config file.
The trained model is valid for 14:2:1 phases only, which are stoichiometric compositions and the predicted Tc value is in Kelvin and at room temperature.
**Model pipeline**
The voting regressor to predict the Tc combines the following four base models and equal weight is assigned to each base model.
1. Extra tree regressor (ET)
2. Extreme gradient boosting (XGB)
3. Random forest regressor (RF)
4. Adaptive boosted RF regressor (AB)
# How to use the trained model for inference
```python
import json
import joblib
import pandas as pd
Tc_predictor = joblib.load('Magnet_Tc_predictor.joblib') # trained model
config = json.load(open('config.json'))
features = config['features'] # feature vector
#data = pd.read_excel("data.xlsx") # read test file with new compositions
data = data[features]
Predicted_value = Tc_predictor.predict(data) # predict Tc values
print("Predicted Tc value is: {0:.2f}'.format(Predicted_value)")
```
|